Spaces:

NN-BRD
/

MMpose

Build error

xmrt commited on Aug 14, 2023

Commit

d567440

1 Parent(s): 59f93ed

no cuda??

Files changed (1) hide show

main.py CHANGED Viewed

@@ -38,7 +38,7 @@ def tracking(video, model, boxes=True):
     # Perform tracking with the model
     print("[INFO] Starting tracking!")
     # https://docs.ultralytics.com/modes/predict/
-    annotated_frame = model(video, device="cuda", boxes=boxes)
     return annotated_frame

     # Perform tracking with the model
     print("[INFO] Starting tracking!")
     # https://docs.ultralytics.com/modes/predict/
+    annotated_frame = model(video, boxes=boxes)
     return annotated_frame