ArvindSelvaraj commited on
Commit
38c59e6
1 Parent(s): c39b548

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -86,7 +86,7 @@ def generate_testcases(user_story):
86
  try:
87
  # Call the Nvidia llama API with the refined prompt
88
  completion = client.chat.completions.create(
89
- model="meta/llama-3.1-70b-instruct", # Using llama3.1 70b model
90
  messages=[
91
  {"role": "user", "content": prompt}
92
  ],
 
86
  try:
87
  # Call the Nvidia llama API with the refined prompt
88
  completion = client.chat.completions.create(
89
+ model="meta/llama-3.1-405b-instruct", # Using llama3.1 405b model
90
  messages=[
91
  {"role": "user", "content": prompt}
92
  ],