Update app.py

app.py

@@ -4,6 +4,8 @@ from fastai.vision.all import *
 from icevision.all import *
 import PIL
 
+class_map = ['raccoon']
+
 model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
 state_dict = torch.load('fasterRCNN_resnet18_Raccoons.pth')
 model1.load_state_dict(state_dict)