fruitpicker01 commited on
Commit
14a7232
1 Parent(s): 8bf8bbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -178,8 +178,8 @@ def generate_message_mistral_generate(prompt, max_retries=5):
178
  try:
179
  chat_response = client_mistral_generate.chat.complete(
180
  model="open-mistral-nemo",
181
- "temperature": 0.8,
182
- "max_tokens": 68,
183
  messages=[
184
  {
185
  "role": "user",
@@ -205,7 +205,7 @@ def generate_message_mistral_check(prompt, max_retries=5):
205
  try:
206
  chat_response = client_mistral_check.chat.complete(
207
  model="mistral-large-latest",
208
- "temperature": 0.2,
209
  messages=[
210
  {
211
  "role": "user",
 
178
  try:
179
  chat_response = client_mistral_generate.chat.complete(
180
  model="open-mistral-nemo",
181
+ temperature=0.8,
182
+ max_tokens=68,
183
  messages=[
184
  {
185
  "role": "user",
 
205
  try:
206
  chat_response = client_mistral_check.chat.complete(
207
  model="mistral-large-latest",
208
+ temperature=0.2,
209
  messages=[
210
  {
211
  "role": "user",