Sergidev commited on
Commit
69b70aa
Β·
verified Β·
1 Parent(s): 50380d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,7 +58,7 @@ def get_model() -> tuple[MMAudio, FeaturesUtils, SequenceConfig]:
58
  net, feature_utils, seq_cfg = get_model()
59
 
60
 
61
- @spaces.GPU(duration=120)
62
  @torch.inference_mode()
63
  def video_to_audio(video: gr.Video, prompt: str, negative_prompt: str, seed: int, num_steps: int,
64
  cfg_strength: float, duration: float):
@@ -98,7 +98,7 @@ def video_to_audio(video: gr.Video, prompt: str, negative_prompt: str, seed: int
98
  return video_save_path
99
 
100
 
101
- @spaces.GPU(duration=120)
102
  @torch.inference_mode()
103
  def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int, cfg_strength: float,
104
  duration: float):
 
58
  net, feature_utils, seq_cfg = get_model()
59
 
60
 
61
+ @spaces.GPU(duration=20)
62
  @torch.inference_mode()
63
  def video_to_audio(video: gr.Video, prompt: str, negative_prompt: str, seed: int, num_steps: int,
64
  cfg_strength: float, duration: float):
 
98
  return video_save_path
99
 
100
 
101
+ @spaces.GPU(duration=20)
102
  @torch.inference_mode()
103
  def text_to_audio(prompt: str, negative_prompt: str, seed: int, num_steps: int, cfg_strength: float,
104
  duration: float):