Increase timeout
Browse files
app.py
CHANGED
|
@@ -92,6 +92,7 @@ def generate(
|
|
| 92 |
client = Client(
|
| 93 |
model2endpoint[model_name],
|
| 94 |
headers={"Authorization": f"Bearer {API_TOKEN}"},
|
|
|
|
| 95 |
)
|
| 96 |
# Don't return meaningless message when the input is empty
|
| 97 |
if not user_message:
|
|
|
|
| 92 |
client = Client(
|
| 93 |
model2endpoint[model_name],
|
| 94 |
headers={"Authorization": f"Bearer {API_TOKEN}"},
|
| 95 |
+
timeout=60,
|
| 96 |
)
|
| 97 |
# Don't return meaningless message when the input is empty
|
| 98 |
if not user_message:
|