Update app.py

app.py

@@ -135,7 +135,14 @@ class FallDetection:
     @spaces.GPU
     def fall_detect(self, video_path):
         try:
+            import torch
+            import os
+            import cv2
+            import numpy as np
+            from ultralytics import YOLO
+
             device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
             if not os.path.exists(self.model_path):
                 model = YOLO("yolov8l.pt")
                 model.save(self.model_path)
@@ -160,8 +167,12 @@ class FallDetection:
                 ret, frame = cap.read()
                 if not ret:
                     break
+                
                 frame = cv2.resize(frame, (width, height))
-                results = model(frame)
+
+                # ✅ FIX: Use `.predict()` to ensure proper YOLO inference
+                results = model.predict(frame, imgsz=640, device=device)  
+
                 for result in results:
                     boxes = result.boxes.xyxy.cpu().numpy()
                     classes = result.boxes.cls.cpu().numpy()