Update app.py

app.py

@@ -49,7 +49,7 @@ h1, .gr-title {color: #00e6d3 !important; font-family: 'Segoe UI', sans-serif;}
 
 demo = gr.Interface(
     fn=process_image,
-    inputs=gr.Image(type="pil", label="Upload Grayscale Landscape", image_mode="L", shape=(256, 256)),
+    inputs=gr.Image(type="pil", label="Upload Grayscale Landscape", image_mode="L"),
     outputs=gr.Image(type="pil", label="Colorized Output"),
     title="🌄 Gray2Color Landscape Autoencoder",
     description=(