NavyDevilDoc commited on
Commit
2bf803c
·
verified ·
1 Parent(s): b2968b1

Update src/app.py

Browse files
Files changed (1) hide show
  1. 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=300)
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})