Update app.py

app.py

@@ -60,8 +60,11 @@ def process_input(input_image, upscale_factor):
         )
         target_input_pixels = MAX_PIXEL_BUDGET / (upscale_factor ** 2)
         scale = (target_input_pixels / (w * h)) ** 0.5
-        new_w = int(w * scale) - int(w * scale) % 8
-        new_h = int(h * scale) - int(h * scale) % 8
+        new_w = int(w * scale) // 16 * 16  # Ensure divisible by 16 for Flux compatibility
+        new_h = int(h * scale) // 16 * 16
+        if new_w == 0 or new_h == 0:
+            new_w = max(16, new_w)
+            new_h = max(16, new_h)
         input_image = input_image.resize((new_w, new_h), resample=Image.LANCZOS)
         was_resized = True
 
@@ -163,7 +166,8 @@ def enhance_image(
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()
-            pipe.enable_model_cpu_offload() if device == "cuda" else None
+            if device == "cuda":
+                pipe.enable_model_cpu_offload()
 
             if USE_ESRGAN:
                 esrgan_path = "4x-UltraSharp.pth"
@@ -230,7 +234,7 @@ def enhance_image(
         control_image,
         denoising_strength,
         num_inference_steps,
-        1.0,  # Hardcoded guidance_scale to 1
+        3.5,  # Updated guidance_scale to match workflow (3.5)
         generator,
         tile_size=1024,
         overlap=32