Update backend.py

backend.py

@@ -6,10 +6,10 @@ import html  # For escaping HTML characters
 from bs4 import BeautifulSoup
 from openai import OpenAI 
 
-# Initialize OpenAI API with Meta's Llama 3.1 405B model
+# Initialize OpenAI API with Nvidia's Llama 3.1 70b nemotron 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):
@@ -46,7 +46,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 llama 3.1 450b
+            model="nvidia/llama-3.1-nemotron-70b-instruct",  # Using llama 3.1 70b
             messages=[
                 {"role": "user", "content": prompt}
             ],