CristopherWVSU commited on
Commit
54ed8be
·
verified ·
1 Parent(s): 227c39f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,13 +3,13 @@ logging.basicConfig(level=logging.DEBUG)
3
 
4
  import streamlit as st
5
 
6
-
7
  from openai import OpenAI
8
 
 
9
  # generateContext = pipeline("text-generation", model="meta-llama/Llama-3.1-405B-Instruct")
10
 
11
  client = OpenAI(base_url = "https://integrate.api.nvidia.com/v1",
12
- api_key = "NVIDIA_API_KEY"
13
  )
14
 
15
  def main():
 
3
 
4
  import streamlit as st
5
 
 
6
  from openai import OpenAI
7
 
8
+ API_Key= os.getenv("NVIDIA_API_KEY")
9
  # generateContext = pipeline("text-generation", model="meta-llama/Llama-3.1-405B-Instruct")
10
 
11
  client = OpenAI(base_url = "https://integrate.api.nvidia.com/v1",
12
+ api_key = API_Key
13
  )
14
 
15
  def main():