Hugging Face's logo

Spaces:
minh9972t12
/
yolocar
Running

App Files Community
minh9972t12 commited on
Commit
4181aa6
·
1 Parent(s): 5045d14

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -44,14 +44,14 @@ MODEL_PATHS = {
44
  2: "models_small_version3/best.pt", # Small v3 PT
45
  3: "models_medium/best.pt", # Medium v1 PT
46
  4: "models_medium_version_2/best.pt", # Medium v2 PT
47
- 5: "models_medium_version3/best.pt", # Medium v3 PT
48
  # ONNX models (optimized with v1.19 + opset 21)
49
  6: "models_small/best.onnx", # Small v1 ONNX
50
  7: "models_small_version_2/best.onnx", # Small v2 ONNX,
51
  8: "models_small_version3/best.onnx", # Small v3 ONNX
52
  9: "models_medium/best.onnx", # Medium v1 ONNX
53
  10: "models_medium_version_2/best.onnx", # Medium v2 ONNX,
54
- 11: "models_medium_version3/best.onnx" # Medium v3 ONNX,
55
  }
56
 
57
  # Config paths - ONNX uses same config as PT version
 
44
  2: "models_small_version3/best.pt", # Small v3 PT
45
  3: "models_medium/best.pt", # Medium v1 PT
46
  4: "models_medium_version_2/best.pt", # Medium v2 PT
47
+ 5: "model_medium_version3/best.pt", # Medium v3 PT
48
  # ONNX models (optimized with v1.19 + opset 21)
49
  6: "models_small/best.onnx", # Small v1 ONNX
50
  7: "models_small_version_2/best.onnx", # Small v2 ONNX,
51
  8: "models_small_version3/best.onnx", # Small v3 ONNX
52
  9: "models_medium/best.onnx", # Medium v1 ONNX
53
  10: "models_medium_version_2/best.onnx", # Medium v2 ONNX,
54
+ 11: "model_medium_version3/best.onnx" # Medium v3 ONNX,
55
  }
56
 
57
  # Config paths - ONNX uses same config as PT version