im size

app.py

@@ -97,9 +97,13 @@ def sepia(input_img):
     fig = draw_plot(pred_img, seg)
     return fig
 
+def proc_img(img):
+    img = img.resize((1024,1024))
+    return img
+
 demo = gr.Interface(fn=sepia,
                     #inputs=gr.Image(shape=(1024, 1024)),
-                    inputs=gr.Image(type='pil', image_mode='RGB', target_size=(1024, 1024)),
+                    inputs=gr.Image(type='pil', prepeocessing_function=proc_img),
                     outputs=['plot'],
                     examples=["city-1c.jpg"],
                     allow_flagging='never')