chong.zhang commited on
Commit
6ed31d5
·
1 Parent(s): e5d1aa9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def trim_audio(audio_file, cut_seconds=5):
102
  torchaudio.save(output_path, cutted_audio, sr)
103
  return output_path
104
 
105
- @spaces.GPU()
106
  def music_generation(args):
107
  set_env_variables()
108
  model = InspireMusicUnified(
 
102
  torchaudio.save(output_path, cutted_audio, sr)
103
  return output_path
104
 
105
+ @spaces.GPU(duration=120)
106
  def music_generation(args):
107
  set_env_variables()
108
  model = InspireMusicUnified(