Fix device='cpu'

app.py

@@ -53,8 +53,7 @@ def predict_video(video_input):
         success, frame = video_reader.read()
 
         if success:
-            # Run YOLO inference on the frame on GPU Device 0
-            results = detection_model(frame, device=0)
+            results = detection_model(frame, device='cpu')
 
             # Visualize the results on the frame
             annotated_frame = results[0].plot()