Update app.py

app.py

@@ -22,7 +22,7 @@ class StopOnTokens(StoppingCriteria):
 
 
 # Function to generate model predictions.
-@spaces.GPU(timeout=420)
+@spaces.GPU(duration=420)
 def predict(message, history):
     history_transformer_format = history + [[message, ""]]
     stop = StopOnTokens()