Update app.py

app.py

@@ -113,15 +113,7 @@ def infer(
     pil_images = []
     for item in [image_1, image_2, image_3]:
         if item is None: continue
-        try:
-            if isinstance(item[0], Image.Image):
-                pil_images.append(item[0].convert("RGB"))
-            elif isinstance(item[0], str):
-                pil_images.append(Image.open(item[0]).convert("RGB"))
-            elif hasattr(item, "name"):
-                pil_images.append(Image.open(item.name).convert("RGB"))
-        except Exception:
-            continue
+        pil_images.append(item.convert("RGB"))
 
     if height==256 and width==256:
         height, width = None, None
@@ -178,9 +170,10 @@ with gr.Blocks(css=css) as demo:
         """)
         with gr.Row():
             with gr.Column():
-                image_1 = gr.Image(label="image 1", type="pil", interactive=True)
-                image_2 = gr.Image(label="image 2", type="pil", interactive=True)
-                image_3 = gr.Image(label="image 3", type="pil", interactive=True)
+                with gr.Row():
+                    image_1 = gr.Image(label="image 1", type="pil", interactive=True)
+                    image_2 = gr.Image(label="image 2", type="pil", interactive=True)
+                    image_3 = gr.Image(label="image 3", type="pil", interactive=True)
 
             with gr.Column():
                 result = gr.Image(label="Result", type="pil", interactive=False)