onnx --> openvino, intel hardware

backend/model.py

@@ -1,3 +1,4 @@
+# Using OpenVino as per Space settings compatible for performance on Intel CPU hardware
 import os
 from pathlib import Path
 from dotenv import load_dotenv
@@ -7,10 +8,10 @@ load_dotenv()
 
 MODEL_DIR = Path(__file__).parent / "weights"
 PT_PATH = MODEL_DIR / "best.pt"
-ONNX_PATH = MODEL_DIR / "best.onnx"
+OV_PATH = MODEL_DIR / "best_openvino_model"  # it's a directory
 
 
-def ensure_onnx():
+def ensure_openvino():
     MODEL_DIR.mkdir(exist_ok=True)
 
     if not PT_PATH.exists():
@@ -21,13 +22,12 @@ def ensure_onnx():
             f'https://huggingface.co/Perception365/VehicleNet-Y26s/resolve/main/weights/best.pt'
         )
 
-    if not ONNX_PATH.exists():
-        YOLO(str(PT_PATH)).export(format="onnx", dynamic=True)
-        exported = PT_PATH.with_suffix(".onnx")
-        if exported != ONNX_PATH:
-            exported.rename(ONNX_PATH)
+    if not OV_PATH.exists():
+        # Export happens relative to the .pt file's location
+        # Result will be: weights/best_openvino_model/
+        YOLO(str(PT_PATH)).export(format="openvino", dynamic=True)
 
 
 def load_model():
-    ensure_onnx()
-    return YOLO(str(ONNX_PATH), task="detect")
+    ensure_openvino()
+    return YOLO(str(OV_PATH), task="detect")