Update app.py

app.py

@@ -43,7 +43,7 @@ model = AutoModelForCausalLM.from_pretrained(
         )
 tokenizer = AutoTokenizer.from_pretrained(MODELS)
 
-@spaces.GPU
+@spaces.GPU(duration=2)
 def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     print(f'message is - {message}')
     print(f'history is - {history}')