Update app.py

app.py

@@ -7,7 +7,7 @@ import PIL
 
 checkpoint_path = "fasterRCNN_resnet18_Raccoons.pth"
 model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True),
-                                       num_classes=1)
+                                       num_classes=2)
 
 state_dict = torch.load(checkpoint_path, map_location=torch.device('cpu'))
 model.load_state_dict(state_dict)