Update app.py

app.py

@@ -5,22 +5,8 @@ import numpy as np
 
 def process_image(input_image):
     # Get image size in KB
-    img_size_kb = os.path.getsize(input_image) / 1024
     
-    # Create output image (300x300 square)
-    size = 300
-    if img_size_kb > 500:
-        # Green square for large images
-        color = [0, 255, 0]  # RGB for green
-    else:
-        # Black square for small images
-        color = [0, 0, 0]  # RGB for black
-    
-    # Create numpy array with the specified color
-    array = np.zeros((size, size, 3), dtype=np.uint8)
-    array[:, :] = color
-    
-    return Image.fromarray(array)
+    return input_image
 
 # Create Gradio interface
 with gr.Blocks() as demo: