Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -131,7 +131,8 @@ class PeopleTracking:
|
|
131 |
class FallDetection:
|
132 |
def __init__(self, yolo_model_path="yolov8l.pt"):
|
133 |
self.model_path = yolo_model_path
|
134 |
-
|
|
|
135 |
def fall_detect(self, video_path):
|
136 |
try:
|
137 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
131 |
class FallDetection:
|
132 |
def __init__(self, yolo_model_path="yolov8l.pt"):
|
133 |
self.model_path = yolo_model_path
|
134 |
+
|
135 |
+
@spaces.GPU
|
136 |
def fall_detect(self, video_path):
|
137 |
try:
|
138 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|