Update app.py

app.py

@@ -47,10 +47,9 @@ def get_figure(in_pil_img, in_results):
     return plt.gcf()
 
 
-def predict(model, in_pil_img):
-
+def predict(in_pil_img):
+    
     results = model(in_pil_img)
-
     figure = get_figure(in_pil_img, results)
 
     buf = io.BytesIO()
@@ -62,7 +61,7 @@ def predict(model, in_pil_img):
 
 
 input = gr.inputs.Image(label="Upload your Image", type = 'pil', optional=True)
-output = gr.outputs.Textbox(label="Captions")
+output = gr.Image(label="Output image with predicted instances", type="pil")
 
 
 interface = gr.Interface(