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 Nvidia's Llama model
+# Initialize OpenAI API with Nvidia's Nematron 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 llama3.1 405b model
+            model="nvidia/nemotron-4-340b-instruct",  # Using nematron
             messages=[
                 {"role": "user", "content": prompt}
             ],