code reviewed

train.py

@@ -36,8 +36,13 @@ def train_model():
         cache='disk',        # Changed to disk caching for deterministic results
     )
     
+     # Create model directory if it doesn't exist
+    import os
+    os.makedirs('model/weights', exist_ok=True)
+
     # Save the trained model
     model.save('model/weights/best.pt')
 
+
 if __name__ == '__main__':
     train_model()