Spaces:
Paused
Paused
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(duration=
|
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()
|