Update app.py

app.py

@@ -178,12 +178,12 @@ def generate_message_mistral_generate(prompt, max_retries=5):
         try:
             chat_response = client_mistral_generate.chat.complete(
                 model="open-mistral-nemo",
+                "temperature": 0.8,
+                "max_tokens": 68,
                 messages=[
                     {
                         "role": "user",
-                        "content": prompt,
-                        "max_tokens": 30,
-                        "temperature": 0.8
+                        "content": prompt
                     },
                 ]
             )
@@ -205,12 +205,11 @@ def generate_message_mistral_check(prompt, max_retries=5):
         try:
             chat_response = client_mistral_check.chat.complete(
                 model="mistral-large-latest",
+                "temperature": 0.2,
                 messages=[
                     {
                         "role": "user",
-                        "content": prompt,
-                        "max_tokens": 30,
-                        "temperature": 0.2
+                        "content": prompt
                     },
                 ]
             )