g

app.py

@@ -8,7 +8,7 @@ tokenizer = GPT2TokenizerFast.from_pretrained(model_id, model_max_length=128)
 
 generator = pipeline("text-generation", model=model_id, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id)
 
-@spaces.GPU
+@spaces.GPU(duration=60)
 def generate_text(prompt, max_length, temperature):
     try:
         generated = generator(