Update app.py

app.py

@@ -379,8 +379,8 @@ def generate_music(
     return recons_mix
 
 
-@spaces.GPU(duration=None)
-def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=15):
+@spaces.GPU(duration=120)
+def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=10):
 
     # Ensure the output folder exists
     output_dir = "./output"
@@ -392,7 +392,6 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=
     # Execute the command
     try:
         music = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments, output_dir=output_dir, cuda_idx=0, max_new_tokens=max_new_tokens)
-        
         return music
     except Exception as e:
         gr.Warning("An Error Occured: " + str(e))