Update app.py

app.py

@@ -210,7 +210,8 @@ def draw_plot(pred_img, seg):
     return fig
 
 def sepia(input_img):
-    img = cv2.imread(input_img).astype(np.float32)
+    img = cv2.imread(input_img)
+    img = cv2.resize(img, (640, 640)).astype(np.float32)
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img_batch = np.expand_dims(img, axis=0)
     img_batch = np.transpose(img_batch, (0, 3, 1, 2))
@@ -239,7 +240,7 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(sepia, 
-                    gr.inputs.Image(type="filepath", shape=(640, 640)), 
+                    gr.inputs.Image(type="filepath"), 
                     outputs=['plot'], 
                     examples=["ADE_val_00000001.jpeg"],
                     allow_flagging='never')