CCockrum commited on
Commit
ab8074b
·
verified ·
1 Parent(s): b256ef1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -10,6 +10,15 @@ from langchain_core.prompts import PromptTemplate
10
  from langchain_core.output_parsers import StrOutputParser
11
  from transformers import pipeline
12
 
 
 
 
 
 
 
 
 
 
13
  # Must be the first Streamlit command!
14
  st.set_page_config(page_title="HAL - NASA ChatBot", page_icon="🚀")
15
 
 
10
  from langchain_core.output_parsers import StrOutputParser
11
  from transformers import pipeline
12
 
13
+ # Use environment variables for keys
14
+ HF_TOKEN = os.getenv("HF_TOKEN")
15
+ if HF_TOKEN is None:
16
+ raise ValueError("HF_TOKEN environment variable not set. Please set it in your Hugging Face Space settings.")
17
+
18
+ NASA_API_KEY = os.getenv("NASA_API_KEY")
19
+ if NASA_API_KEY is None:
20
+ raise ValueError("NASA_API_KEY environment variable not set. Please set it in your Hugging Face Space settings.")
21
+
22
  # Must be the first Streamlit command!
23
  st.set_page_config(page_title="HAL - NASA ChatBot", page_icon="🚀")
24