rwayz commited on
Commit
8115357
·
verified ·
1 Parent(s): d60771a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def chatbot_response(user_input, model_name):
38
  completion = client.chat.completions.create(
39
  model=model_id,
40
  messages=messages,
41
- max_tokens=8192 if any(m in model_id for m in ["Qwen", "DeepSeek"]) else 800
42
  )
43
  response = completion.choices[0].message['content']
44
  except Exception as e:
 
38
  completion = client.chat.completions.create(
39
  model=model_id,
40
  messages=messages,
41
+ max_tokens=8192 if "Qwen" in model_id else 900,
42
  )
43
  response = completion.choices[0].message['content']
44
  except Exception as e: