mkhodary101 commited on
Commit
16b5fca
·
verified ·
1 Parent(s): 1a67c1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- spaces@GPU
 
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")