modified: app.py

app.py

@@ -116,7 +116,7 @@ model = AutoModelForCausalLM.from_pretrained(
 model.to(device)
 model.eval()
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=120)
 def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=200):
 
     # Ensure the output folder exists