Update firedetection.py

firedetection.py

@@ -8,7 +8,8 @@ def detect_fire_in_video(
     input_video_path: str,
     output_video_path: str,
     model_path: str,
-    device: str = "cpu"
+    model.to('cuda' if torch.cuda.is_available() else 'cpu')
+
 ) -> str:
     """
     Detects fire in the given video using a YOLO model.