SonFox2920 commited on
Commit
1cede7c
·
verified ·
1 Parent(s): 8fb47d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def load_models():
50
  """Load both object detection and classification models"""
51
  # Load object detection model
52
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
53
- object_detection_model = torch.load("fasterrcnn_resnet50_fpn_270824.pth", map_location=device)
54
  object_detection_model.to(device)
55
  object_detection_model.eval()
56
 
 
50
  """Load both object detection and classification models"""
51
  # Load object detection model
52
  device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
53
+ object_detection_model = torch.load("fasterrcnn_resnet50_fpn_090824.pth", map_location=device)
54
  object_detection_model.to(device)
55
  object_detection_model.eval()
56