Add YOLO("./model/weights/best.pt")

joshbickett · joshbickett · commit bc831f8dd42e · 2024-01-07T08:48:27.000-08:00
diff --git a/operate/actions.py b/operate/actions.py
@@ -40,7 +40,8 @@
 
 client = config.initialize_openai_client()
 
-yolo_model = None
+
+yolo_model = YOLO("./model/weights/best.pt")  # Load your trained model
 
 
 async def get_next_action(model, messages, objective):
