Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ SEARXNG_KEY = 'f9f07f93b37b8483aadb5ba717f556f3a4ac507b281b4ca01e6c6288aa3e3ae5'
|
|
| 46 |
# Use the environment variable
|
| 47 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 48 |
client = InferenceClient(
|
| 49 |
-
"mistralai/Mistral-
|
| 50 |
token=HF_TOKEN,
|
| 51 |
)
|
| 52 |
|
|
@@ -396,7 +396,7 @@ Instructions:
|
|
| 396 |
return response.choices[0].message.content.strip()
|
| 397 |
elif model == "mistral":
|
| 398 |
response = mistral_client.chat.complete(
|
| 399 |
-
model="
|
| 400 |
messages=messages,
|
| 401 |
max_tokens=10000,
|
| 402 |
temperature=temperature,
|
|
|
|
| 46 |
# Use the environment variable
|
| 47 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 48 |
client = InferenceClient(
|
| 49 |
+
"mistralai/Mistral-Small-Instruct-2409",
|
| 50 |
token=HF_TOKEN,
|
| 51 |
)
|
| 52 |
|
|
|
|
| 396 |
return response.choices[0].message.content.strip()
|
| 397 |
elif model == "mistral":
|
| 398 |
response = mistral_client.chat.complete(
|
| 399 |
+
model="open-mistral-nemo",
|
| 400 |
messages=messages,
|
| 401 |
max_tokens=10000,
|
| 402 |
temperature=temperature,
|