Update app.py

app.py

@@ -172,19 +172,22 @@ def infer(
         num_images_per_prompt=1,
     ).images[0]
 
-    # *** FIX: Changed function to return only 2 values to match the button's expectation ***
     return result, seed
 
 # --- Wrapper for Examples ---
 @spaces.GPU
-def infer_example(input_image_path, prompt, lora_adapter):
-    # *** FIX: Fully implemented this function to handle examples correctly ***
-    input_pil = Image.open(input_image_path).convert("RGB")
+def infer_example(input_image, prompt, lora_adapter):
+    # *** FIX: The input is already a PIL Image object from Gradio, not a path. ***
+    # We no longer need Image.open() here.
+    input_pil = input_image.convert("RGB")
+    
     # Calculate aspect ratio for the example image
     width, height = update_dimensions_on_upload(input_pil)
-    # Set reasonable default values for example inference
+    
+    # Set reasonable default values for example inference to get good results
     guidance_scale = 1.0
     steps = 4
+    
     # Call the main infer function
     result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps, width, height)
     return result, seed
@@ -250,7 +253,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed],
             fn=infer_example,
-            cache_examples="lazy", # Changed to lazy for better performance
+            cache_examples="lazy",
             label="Examples"
         )