Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|