Update app.py

app.py

@@ -6,6 +6,14 @@ import time
 import os
 import onnxruntime
 from pathlib import Path
+from ultralytics import YOLO
+
+# Load YOLOv5 model for ONNX export
+model = YOLO("yolov5n.pt")  # Use "yolov5x.pt" if you want the larger model
+
+# Export to ONNX format
+model.export(format="onnx", dynamic=True)
+
 
 os.makedirs("models", exist_ok=True)