0x7o commited on
Commit
ec93e16
1 Parent(s): 9d1bb91

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
26
  def predict(message, history):
27
  stop = StopOnTokens()
28
  conversation = []
 
22
 
23
 
24
  # Function to generate model predictions.
25
+ @spaces.GPU(duration=600)
26
  def predict(message, history):
27
  stop = StopOnTokens()
28
  conversation = []