Update run.py

run.py

@@ -1,18 +1,18 @@
 import gradio as gr
 
-def process_images(image1, image2):
-    # This function will just return the two images for demonstration purposes
+def process_images(image_list):
+    # Assume image_list is a list containing two images
+    image1, image2 = image_list
+    # For demonstration, let's just return the two images
     return image1, image2
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        file_input_1 = gr.Image()
-        file_input_2 = gr.Image()
+    file_input = gr.File(label="Upload two images", file_count="multiple", type="file")
     output_1 = gr.Image()
     output_2 = gr.Image()
     btn = gr.Button("Submit")
 
-    btn.click(process_images, inputs=[file_input_1, file_input_2], outputs=[output_1, output_2])
+    btn.click(process_images, inputs=file_input, outputs=[output_1, output_2])
 
 if __name__ == "__main__":
     demo.launch()