Hugging Face's logo

Spaces:
tryolabs
/
norfair-demo
Runtime error

App Files Community
1
Diego Fernandez commited on
Commit
4d08d2b
•
1 Parent(s): 3647577

fix: move to cpu

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -46,7 +46,7 @@ def inference(
46
  coord_transformations = None
47
  paths_drawer = None
48
  track_points = Style[track_points].value
49
- model = YOLO(ModelsPath[model].value, device="cuda")
50
  video = Video(input_path=input_video, output_path="tmp")
51
 
52
  if motion_estimation:
 
46
  coord_transformations = None
47
  paths_drawer = None
48
  track_points = Style[track_points].value
49
+ model = YOLO(ModelsPath[model].value)
50
  video = Video(input_path=input_video, output_path="tmp")
51
 
52
  if motion_estimation: