Update app.py

app.py

@@ -16,11 +16,16 @@ hf = HuggingFaceHubEmbeddings(
     huggingfacehub_api_token= token,
 )
 
-from langchain.embeddings import HuggingFaceInferenceAPIEmbeddings
+# from langchain.embeddings import HuggingFaceInferenceAPIEmbeddings
 
-embeddings = HuggingFaceInferenceAPIEmbeddings(
-    api_key=token, model_name="sentence-transformers/all-MiniLM-l6-v2"
-)
+# embeddings = HuggingFaceInferenceAPIEmbeddings(
+#     api_key=token, model_name="sentence-transformers/all-MiniLM-l6-v2"
+# )
+# hf = HuggingFaceHubEmbeddings(
+#     repo_id=repo_id,
+#     task="feature-extraction",
+#     huggingfacehub_api_token= HUGGINGFACEHUB_API_TOKEN,
+# )
 
 
 def main():
@@ -52,7 +57,7 @@ def main():
       # embeddings = query(chunks)
       # embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
       
-      knowledge_base = FAISS.from_texts(chunks, embeddings)
+      knowledge_base = FAISS.from_texts(chunks, hf)
       
       # show user input
       user_question = st.text_input("Ask a question about your PDF:")