Update backend.py

backend.py

@@ -9,7 +9,7 @@ from openai import OpenAI
 # Initialize OpenAI API with Nvidia's llama 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 llama API with the refined prompt
         completion = client.chat.completions.create(
-            model="meta/llama-3.1-405b-instruct",  # Using llama3.1 405b model
+            model="meta/llama-3.1-70b-instruct",  # Using llama3.1 70b model
             messages=[
                 {"role": "user", "content": prompt}
             ],