SophieNgo commited on
Commit
2cf6d1b
·
verified ·
1 Parent(s): 162bb86

Update app.py

Browse files

we switch to cpu again

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ from PromptTrack import PromptTracker
22
  tracker = PromptTracker()
23
 
24
 
25
- @spaces.GPU(duration=300)
26
  def process_video(video_path, prompt):
27
  import torch
28
  print(f"Is CUDA available: {torch.cuda.is_available()}")
 
22
  tracker = PromptTracker()
23
 
24
 
25
+ #@spaces.GPU(duration=300)
26
  def process_video(video_path, prompt):
27
  import torch
28
  print(f"Is CUDA available: {torch.cuda.is_available()}")