[fix error]

app.py

@@ -64,14 +64,17 @@ sample_images = [
 
 iface = gr.Interface(
     fn=reflection_removal,
-    inputs=gr.File(file_types=["image"], label="Upload Image (JPG/PNG)", live=True),
-    outputs=gr.Image(type="pil", label="Processed Image"),
-    title="Reflection Remover with Pix2Pix",
-    description="Upload a single image to remove reflections. No GIFs allowed.",
+    inputs=[
+        gr.Image(source="upload", type="filepath", label="Upload Image (JPG/PNG)", live=True)
+    ],
+    outputs=gr.Gallery(label="Results after Reflection Removal"),
     examples=[
-        os.path.join(SAMPLE_DIR, img) for img in sample_images
+        os.path.join("sample_images", img) for img in os.listdir("sample_images") if img.endswith((".jpg", ".jpeg", ".png"))
     ],
+    title="Reflection Remover with Pix2Pix",
+    description="Upload images to remove reflections using a Pix2Pix model. You can also try the sample images below."
 )
 
 if __name__ == "__main__":
     iface.launch()
+