Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = []
|