Shreyas094
commited on
Commit
•
d0bc86a
1
Parent(s):
00a13ce
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,
|