Spaces:
Running
Running
bainskarman
commited on
Update model.py
Browse files
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):
|