Test_Pipeline_dev

Sleeping

fruitpicker01 commited on Oct 14, 2024

Commit

02373fa

•

1 Parent(s): b121597

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,7 +182,7 @@ def generate_message_mistral_generate(prompt, max_retries=5):
                     {
                         "role": "user",
                         "content": prompt,
-                        "max_tokens": 68,
                         "temperature": 0.8
                     },
                 ]
@@ -209,7 +209,7 @@ def generate_message_mistral_check(prompt, max_retries=5):
                     {
                         "role": "user",
                         "content": prompt,
-                        "max_tokens": 3000,
                         "temperature": 0.2
                     },
                 ]

                     {
                         "role": "user",
                         "content": prompt,
+                        "max_tokens": 30,
                         "temperature": 0.8
                     },
                 ]
                     {
                         "role": "user",
                         "content": prompt,
+                        "max_tokens": 30,
                         "temperature": 0.2
                     },
                 ]