Update app.py

app.py

@@ -5,8 +5,9 @@ from icevision.all import *
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
 repo_id = "inigo99/kangaroo-detector"
 
+class_map = ClassMap(['kangaroo'])
 model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True),
-                                             num_classes=1)
+                                             num_classes=len(class_map))
 state_dict = torch.load("fasterRCNNkangaroo.pth")
 model.load_state_dict(state_dict)
 
@@ -18,5 +19,5 @@ def predict(img):
     return pred_dict['img']
     
 # Creamos la interfaz y la lanzamos. 
-gr.Interface(fn=predict, inputs=gr.inputs.Image(), examples=['00001.jpg','00002.jpg']).launch(share=False)
+gr.Interface(fn=predict, inputs=gr.inputs.Image(size=(256, 256)), examples=['00001.jpg','00002.jpg']).launch(share=False)