Rex Cheng commited on
Commit
627e0b8
1 Parent(s): 35d9519
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -95,6 +95,7 @@ def video_to_audio(video: gr.Video, prompt: str, negative_prompt: str, seed: int
95
  return video_save_path
96
 
97
 
 
98
  @torch.inference_mode()
99
  def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int, cfg_strength: float,
100
  duration: float):
 
95
  return video_save_path
96
 
97
 
98
+ @spaces.GPU
99
  @torch.inference_mode()
100
  def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int, cfg_strength: float,
101
  duration: float):