Walmart-the-bag commited on
Commit
87a7979
1 Parent(s): fdcc962

more seconds

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ class StopOnTokens(StoppingCriteria):
19
  return True
20
  return False
21
 
22
- @spaces.GPU(120)
23
  def predict(message, history, temperature, max_tokens, top_p, top_k):
24
  history_transformer_format = history + [[message, ""]]
25
  stop = StopOnTokens()
 
19
  return True
20
  return False
21
 
22
+ @spaces.GPU(duration=120)
23
  def predict(message, history, temperature, max_tokens, top_p, top_k):
24
  history_transformer_format = history + [[message, ""]]
25
  stop = StopOnTokens()