liuhuadai commited on
Commit
f844705
·
verified ·
1 Parent(s): 0e58869

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -241,8 +241,9 @@ def get_video_duration(video_path):
241
  video = VideoFileClip(video_path)
242
  return video.duration
243
 
244
- @spaces.GPU(duration=360)
245
  @torch.inference_mode()
 
246
  def get_audio(video_path, caption):
247
  # 允许caption为空
248
  if caption is None:
 
241
  video = VideoFileClip(video_path)
242
  return video.duration
243
 
244
+ @spaces.GPU(duration=200)
245
  @torch.inference_mode()
246
+ @torch.no_grad()
247
  def get_audio(video_path, caption):
248
  # 允许caption为空
249
  if caption is None: