Update app.py

app.py

@@ -14,7 +14,8 @@ train_tfms = tfms.A.Adapter(
 )
 valid_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(size), tfms.A.Normalize()])
 
-model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True), 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))
+model1 = faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True), num_classes=len(class_map))
 state_dict = torch.load('fasterRCNNFruits.pth', map_location=torch.device('cpu'))
 model1.load_state_dict(state_dict)