Update app.py

app.py

@@ -6,7 +6,7 @@ import PIL
 
 class_map = ['raccoon']
 
-model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
+model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True), num_classes=len(class_map))
 state_dict = torch.load('fasterRCNN_resnet18_Raccoons.pth')
 model1.load_state_dict(state_dict)