Spaces:

ArvindSelvaraj
/

QA_UserStory_TestCase_Generator

Running

ArvindSelvaraj commited on Sep 13

Commit

cbbeb85

•

1 Parent(s): a477e6a

Update backend.py

Files changed (1) hide show

backend.py CHANGED Viewed

@@ -86,7 +86,7 @@ def generate_testcases(user_story):
     try:
         # Call the Nvidia Mistral API with the refined prompt
         completion = client.chat.completions.create(
-            model="llama3.1-70b",  # Using Mistral model
             messages=[
                 {"role": "user", "content": prompt}
             ],

     try:
         # Call the Nvidia Mistral API with the refined prompt
         completion = client.chat.completions.create(
+            model="llama3.1-405b",  # Using Mistral model
             messages=[
                 {"role": "user", "content": prompt}
             ],