Update generator.py

generator.py

@@ -1,7 +1,12 @@
 import os
+import streamlit as st
 from langchain_community.llms import HuggingFaceEndpoint
 from pydantic import ValidationError
 
+st.write(
+	"Has environment variables been set:",
+	os.environ["HUGGINGFACE_API_TOKEN"] == st.secrets["HUGGINGFACE_API_TOKEN"])
+
 def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2", token=os.getenv("HUGGINGFACE_API_TOKEN")):
     '''
     Load the LLM from the HuggingFace model hub