Update backend.py

backend.py

@@ -9,7 +9,7 @@ from openai import OpenAI
 # Initialize OpenAI API with Nvidia's Mistral model
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
-    api_key="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"
+    api_key="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"
 )
 
 def clean_test_case_output(text):
@@ -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="nv-mistralai/mistral-nemo-12b-instruct",  # Using Mistral model
+            model="meta/llama-3.1-405b-instruct",  # Using Mistral model
             messages=[
                 {"role": "user", "content": prompt}
             ],