Update app.py

app.py

@@ -19,7 +19,7 @@ def load_model( backbone_name, num_classes):
         model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(pretrained=False)
         in_features = model.roi_heads.box_predictor.cls_score.in_features
         model.roi_heads.box_predictor = FastRCNNPredictor(in_features, num_classes)
-        model.load_state_dict(torch.load("fasterrcnnMobilenet", map_location=device))
+        model.load_state_dict(torch.load("fasterrcnnMobilenet.pth", map_location=device))
     model.to(device)
     model.eval()
     return model