TanelAlumae commited on
Commit
7791c48
·
verified ·
1 Parent(s): 6722e17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -61,10 +61,9 @@ def convert_to_vtt(whisper_output):
61
 
62
  def dynamic_gpu_duration(func, duration, *args):
63
 
64
- # @torch.inference_mode()
65
  @spaces.GPU(duration=duration)
66
  def wrapped_func():
67
- yield from func(*args)
68
 
69
  return wrapped_func()
70
 
 
61
 
62
  def dynamic_gpu_duration(func, duration, *args):
63
 
 
64
  @spaces.GPU(duration=duration)
65
  def wrapped_func():
66
+ return from func(*args)
67
 
68
  return wrapped_func()
69