Update app.py

app.py

@@ -5,7 +5,8 @@ import gradio as gr
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(img):
     class_map = ClassMap(['Kangaroo'])
-    model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
+    model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True), num_classes=len(class_map))
+    # model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
     state_dict = torch.load('model.pth')
     model.load_state_dict(state_dict)