bainskarman commited on
Commit
6239656
·
verified ·
1 Parent(s): b0c8c8c

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -9,7 +9,7 @@ from openai import OpenAI
9
  # Initialize OpenAI API with Nvidia's Llama 3.1 70b nemotron model
10
  client = OpenAI(
11
  base_url="https://integrate.api.nvidia.com/v1",
12
- api_key="KEY"
13
  )
14
 
15
  def clean_text_output(text):
 
9
  # Initialize OpenAI API with Nvidia's Llama 3.1 70b nemotron model
10
  client = OpenAI(
11
  base_url="https://integrate.api.nvidia.com/v1",
12
+ api_key=os.environ.get("KEY")
13
  )
14
 
15
  def clean_text_output(text):