Update app.py

app.py

@@ -65,6 +65,8 @@ pipe = FluxImg2ImgPipeline.from_pretrained(
     torch_dtype=torch.bfloat16
 )
 pipe.to(device)
+pipe.enable_vae_tiling()
+pipe.enable_vae_slicing()
 
 print("✅ All models loaded successfully!")
 
@@ -117,7 +119,8 @@ def process_input(input_image, upscale_factor):
             (
                 int(aspect_ratio * MAX_PIXEL_BUDGET**0.5 // upscale_factor),
                 int(MAX_PIXEL_BUDGET**0.5 // aspect_ratio // upscale_factor),
-            )
+            ),
+            resample=Image.LANCZOS
         )
         was_resized = True
 
@@ -126,7 +129,7 @@ def process_input(input_image, upscale_factor):
     w = w - w % 8
     h = h - h % 8
 
-    return input_image.resize((w, h)), w_original, h_original, was_resized
+    return input_image.resize((w, h), resample=Image.LANCZOS), w_original, h_original, was_resized
 
 
 def load_image_from_url(url):
@@ -180,9 +183,9 @@ def enhance_image(
     else:
         prompt = custom_prompt if custom_prompt.strip() else ""
 
-    # Rescale with upscale factor
+    # Rescale with upscale factor using LANCZOS
     w, h = input_image.size
-    control_image = input_image.resize((w * upscale_factor, h * upscale_factor))
+    control_image = input_image.resize((w * upscale_factor, h * upscale_factor), resample=Image.LANCZOS)
 
     generator = torch.Generator().manual_seed(seed)
 
@@ -204,7 +207,7 @@ def enhance_image(
         gr.Info(f"📏 Resizing output to target size: {w_original * upscale_factor}x{h_original * upscale_factor}")
 
     # Resize to target desired size
-    final_image = image.resize((w_original * upscale_factor, h_original * upscale_factor))
+    final_image = image.resize((w_original * upscale_factor, h_original * upscale_factor), resample=Image.LANCZOS)
     
     return [true_input_image, final_image], seed, generated_caption if use_generated_caption else ""
 
@@ -268,7 +271,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 minimum=8,
                 maximum=50,
                 step=1,
-                value=28,
+                value=25,
                 info="More steps = better quality but slower"
             )
             
@@ -287,7 +290,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 maximum=1.0,
                 step=0.05,
                 value=0.3,
-                info="Controls how much the image is transformed (from Ultimate SD Upscaler concept)"
+                info="Controls how much the image is transformed"
             )
             
             with gr.Row():
@@ -337,8 +340,8 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
     # Examples
     gr.Examples(
         examples=[
-            [None, "https://upload.wikimedia.org/wikipedia/commons/thumb/a/a7/Example.jpg/800px-Example.jpg", 42, False, 28, 2, 3.5, 0.3, True, ""],
-            [None, "https://picsum.photos/512/512", 123, False, 25, 3, 4.0, 0.4, True, ""],
+            [None, "https://upload.wikimedia.org/wikipedia/commons/thumb/a/a7/Example.jpg/800px-Example.jpg", 42, False, 25, 2, 3.5, 0.3, True, ""],
+            [None, "https://picsum.photos/512/512", 123, False, 25, 3, 4.0, 0.3, True, ""],
         ],
         inputs=[
             input_image,
@@ -377,10 +380,10 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
         <h4>💡 How it works:</h4>
         <ol>
             <li><strong>Florence-2</strong> analyzes your image and generates a detailed caption</li>
-            <li><strong>FLUX Img2Img</strong> uses this caption to guide the upscaling process with denoising</li>
-            <li>The result is an enhanced, higher-resolution image with improved details</li>
+            <li>Initial upscale with LANCZOS interpolation</li>
+            <li><strong>FLUX Img2Img</strong> enhances the upscaled image with AI diffusion guided by the caption</li>
         </ol>
-        <p><strong>Note:</strong> Due to memory constraints, output is limited to 1024x1024 pixels total budget.</p>
+        <p><strong>Note:</strong> Output limited to 4096x4096 pixels total budget to prevent memory issues.</p>
     </div>
     """)