Shreyas094 commited on
Commit
d0bc86a
1 Parent(s): 00a13ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-Nemo-Instruct-2407",
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="Mistral-Nemo-Instruct-2407",
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,