Update app.py

app.py

@@ -4,9 +4,13 @@ from fastai.vision.all import *
 from icevision.all import *
 
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
-repo_id = "igmarco/raccoon-image-detection"
+# repo_id = "igmarco/raccoon-image-detection"
 
-model1 = from_pretrained_fastai(repo_id)
+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)
+
+# model1 = from_pretrained_fastai(repo_id)
 
 def show_preds(input_image, display_label, display_bbox, detection_threshold):