Spaces:
Running
Running
Update app.py
Browse files
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 "Qwen"
|
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 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:
|