Update main.py

main.py

@@ -1,3 +1,14 @@
+from ultralytics import YOLO
+
+# Load a model
+#model = YOLO('yolov8n-cls.pt')  # load an official model
+model = YOLO('best.pt')  # load a custom model
+
+# Predict with the model
+results = model('bus.jpg') 
+print(results)# predict on an image
+
+
 import asyncio
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware