Update app.py

app.py

@@ -89,7 +89,7 @@ def infer(input_images, prompt, seed=42, randomize_seed=False, guidance_scale=2.
         input_images = [input_images]
     
     # Filter out None images
-    valid_images = [img for img in input_images if img is not None]
+    valid_images = [img[0] for img in input_images if img is not None]
     
     if not valid_images:
         raise gr.Error("Please upload at least one valid image.")
@@ -147,7 +147,8 @@ with gr.Blocks(css=css) as demo:
                     rows=2,
                     object_fit="contain",
                     height="auto",
-                    type="pil"
+                    file_types=['image'],
+                    type='pil'
                 )