Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def load_model(file_path):
|
|
13 |
model = project.version(3).model
|
14 |
|
15 |
# Load the model weights into a PyTorch model
|
16 |
-
pytorch_model = YOLO('
|
17 |
pytorch_model.load_state_dict(torch.load(file_path, map_location=torch.device('cpu')))
|
18 |
pytorch_model.eval()
|
19 |
|
|
|
13 |
model = project.version(3).model
|
14 |
|
15 |
# Load the model weights into a PyTorch model
|
16 |
+
pytorch_model = YOLO('data.yaml')
|
17 |
pytorch_model.load_state_dict(torch.load(file_path, map_location=torch.device('cpu')))
|
18 |
pytorch_model.eval()
|
19 |
|