add specific hf repo ids

chat/bot.py

@@ -22,16 +22,18 @@ load_dotenv()
 PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
 PINECONE_INDEX = os.getenv("PINECONE_INDEX")
 HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-EMBEDDINGS_MODEL = os.getenv("EMBEDDINGS_MODEL")
-CHAT_MODEL = os.getenv("CHAT_MODEL")
+# EMBEDDINGS_MODEL = os.getenv("EMBEDDINGS_MODEL")
+# CHAT_MODEL = os.getenv("CHAT_MODEL")
+EMBEDDINGS_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
+CHAT_MODEL = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
 # Supplement with streamlit secrets if None
-if None in [PINECONE_API_KEY, PINECONE_INDEX, HUGGINGFACE_API_TOKEN, EMBEDDINGS_MODEL, CHAT_MODEL]:
+if None in [PINECONE_API_KEY, PINECONE_INDEX, HUGGINGFACE_API_TOKEN]: #, EMBEDDINGS_MODEL, CHAT_MODEL]:
     PINECONE_API_KEY = st.secrets["PINECONE_API_KEY"]
     PINECONE_INDEX = st.secrets["PINECONE_INDEX"]
     HUGGINGFACE_API_TOKEN = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
-    EMBEDDINGS_MODEL = st.secrets["EMBEDDINGS_MODEL"]
-    CHAT_MODEL = st.secrets["CHAT_MODEL"]
+    # EMBEDDINGS_MODEL = st.secrets["EMBEDDINGS_MODEL"]
+    # CHAT_MODEL = st.secrets["CHAT_MODEL"]
 
 def ChatBot():
     embeddings = HuggingFaceEmbeddings(model_name=EMBEDDINGS_MODEL)