0x7o commited on
Commit
23481c6
1 Parent(s): dc59912

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ class StopOnTokens(StoppingCriteria):
22
 
23
 
24
  # Function to generate model predictions.
25
- @spaces.GPU(duration=420)
26
  def predict(message, history):
27
  history_transformer_format = history + [[message, ""]]
28
  stop = StopOnTokens()
 
22
 
23
 
24
  # Function to generate model predictions.
25
+ @spaces.GPU(duration=60)
26
  def predict(message, history):
27
  history_transformer_format = history + [[message, ""]]
28
  stop = StopOnTokens()