Update app.py

app.py

@@ -1,23 +1,17 @@
 import gradio as gr
-from PIL import Image
-import os
-import numpy as np
 
-def process_image(input_image):
-    # Get image size in KB
-    
-    return input_image
-
-# Create Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## Image Size Checker")
-    gr.Markdown("Upload an image. Output will be green if >500KB, black otherwise.")
-    
+    gr.Markdown("# Image Display Demo")
     with gr.Row():
-        input_image = gr.Image(type="filepath", label="Upload Image")
-        output_image = gr.Image(label="Result", width=300, height=300)
+        with gr.Column():
+            img_input = gr.Image(type="pil", label="Upload Image")
+        with gr.Column():
+            img_output = gr.Image(label="Output Image")
     
-    submit_btn = gr.Button("Check Size")
-    submit_btn.click(fn=process_image, inputs=input_image, outputs=output_image)
+    img_input.change(
+        fn=lambda x: x,
+        inputs=img_input,
+        outputs=img_output
+    )
 
 demo.launch()