satvs commited on
Commit
9c91ae6
·
1 Parent(s): aa755c2

[fix] fix model path for loading

Browse files
Files changed (1) hide show
  1. tasks/image.py +1 -1
tasks/image.py CHANGED
@@ -102,7 +102,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
102
  import torch
103
 
104
  # Load model
105
- model_path = "models"
106
  model_name = "best.pt"
107
  model = YOLO(Path(model_path, model_name))
108
  threshold = 0.14
 
102
  import torch
103
 
104
  # Load model
105
+ model_path = Path("tasks", "models")
106
  model_name = "best.pt"
107
  model = YOLO(Path(model_path, model_name))
108
  threshold = 0.14