Update app.py

app.py

@@ -13,4 +13,4 @@ def predict(img):
    pred_dict = models.torchvision.faster_rcnn.end2end_detect(img, infer_tfms, model.to("cpu"), class_map=class_map, detection_threshold=0.5)
    return pred_dict
    
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128,128)), outputs=gr.outputs.Label(num_top_classes=3),examples=['00001.jpg','00002.jpg']).launch(share=False)
+gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128,128)), outputs=gr.outputs.Image(shape=(128,128),examples=['00001.jpg','00002.jpg']).launch(share=False)