Spaces:
Sleeping
Sleeping
Update src/app.py
Browse files- src/app.py +1 -1
src/app.py
CHANGED
|
@@ -103,7 +103,7 @@ def query_model_universal(messages, max_tokens, model_choice, user_key=None):
|
|
| 103 |
hist += "Assistant: "
|
| 104 |
|
| 105 |
try:
|
| 106 |
-
r = requests.post(url, json={"text": hist, "persona": sys_msg, "max_tokens": max_tokens, "model": tech_name}, timeout=
|
| 107 |
if r.status_code == 200:
|
| 108 |
d = r.json()
|
| 109 |
return d.get("response", ""), d.get("usage", {"input":0,"output":0})
|
|
|
|
| 103 |
hist += "Assistant: "
|
| 104 |
|
| 105 |
try:
|
| 106 |
+
r = requests.post(url, json={"text": hist, "persona": sys_msg, "max_tokens": max_tokens, "model": tech_name}, timeout=600)
|
| 107 |
if r.status_code == 200:
|
| 108 |
d = r.json()
|
| 109 |
return d.get("response", ""), d.get("usage", {"input":0,"output":0})
|