Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
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 |
|