Update app.py

app.py

@@ -57,7 +57,7 @@ def fast_show_mask(annotation, ax):
 
 # post_process(results[0].masks, Image.open("../data/cake.png"))
 
-def predict(input, input_size):
+def predict(input, input_size=512):
     input_size = int(input_size)  # 确保 imgsz 是整数
     results = model(input, device='cpu', retina_masks=True, iou=0.7, conf=0.25, imgsz=input_size)
     pil_image = fast_process(annotations=results[0].masks.data.numpy(), image=input)
@@ -71,7 +71,7 @@ def predict(input, input_size):
 # post_process(annotations=results, image_path=inp)
 
 demo = gr.Interface(fn=predict,
-                    inputs=[gr.inputs.Image(type='pil'), gr.inputs.Dropdown(choices=[512, 800, 1024], default=1024)],
+                    inputs=[gr.inputs.Image(type='pil'), gr.inputs.Dropdown(choices=[512, 800, 1024], default=512)],
                     outputs=['plot'],
                     examples=[["assets/sa_8776.jpg", 1024]],
                     #    ["assets/sa_1309.jpg", 1024]],