Spaces:

Luigi
/

Video-Human-Fall-Detection-with-CLIP

Running on Zero

Luigi commited on 20 days ago

Commit

1268dfc

1 Parent(s): 6a22aef

bugfix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ MODEL_NAME = "openai/clip-vit-base-patch32"
 fall_prompt = "A person falling on the ground."
 nofall_prompt = "A person standing or walking."
-if torch.cuda.is_available():
-    text_inputs = {k: v.cuda() for k, v in text_inputs.items()}
 def extract_frames(video_path, target_size=(224, 224)):
     """

 fall_prompt = "A person falling on the ground."
 nofall_prompt = "A person standing or walking."
+# if torch.cuda.is_available():
+#     text_inputs = {k: v.cuda() for k, v in text_inputs.items()}
 def extract_frames(video_path, target_size=(224, 224)):
     """