Update app.py

app.py

@@ -37,8 +37,8 @@ def get_masks(model_type, image):
 
 iface = gr.Interface(
     fn=get_masks,
-    inputs=[gr.inputs.Image(type="pil"), gr.inputs.Dropdown(['vit_h', 'vit_b', 'vit_l'], label="Model Type")],
-    outputs=gr.outputs.Image(type="pil"),
+    inputs=["image", gr.components.Dropdown(choices=['vit_h', 'vit_b', 'vit_l'], label="Model Type")],
+    outputs="image",
     title="SAM Model Segmentation and Classification",
     description="Upload an image, select a model type, and receive the segmented and classified parts."
 )