rombodawg commited on
Commit
0bf97d4
·
verified ·
1 Parent(s): 9c5a9ef

Update app.py

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