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 |
|