Update model_path in RoDeV1 endpoint to use version 2

Erik172 · May 4, 2024 · 52d7bd0 · 52d7bd0
1 parent 69d0bc8
commit 52d7bd0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/backend/models/rode_v2.pt b/backend/models/rode_v2.pt
diff --git a/backend/resources/RoDe.py b/backend/resources/RoDe.py
@@ -15,7 +15,7 @@ class RoDeV1(Resource):
     def post(self):
         with sentry_sdk.metrics.timing(key="RoDeV1", tags={"model": "RoDeV1"}):
             start_time = datetime.now()
-            model_path = 'models/rode_v1.5.pt'
+            model_path = 'models/rode_v2.pt'
 
             model = YOLO(model_path, verbose=True)