Update app.py

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 class_map=ClassMap(['kangaroo'])
 	
 # Cargamos el learner
-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))
 state_dict = torch.load('fasterRCNNkangaroo.pth')
 model.load_state_dict(state_dict)