rwayz commited on
Commit
30df920
·
verified ·
1 Parent(s): f184505

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(x in model_id for x in ["Qwen", "DeepSeek"]) else 900
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: