SPBhai commited on
Commit
60fb3f3
·
verified ·
1 Parent(s): 5618d7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -9,6 +9,14 @@ from langchain.llms import HuggingFacePipeline
9
  from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
10
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, pipeline
11
 
 
 
 
 
 
 
 
 
12
  # Load the pre-existing vector store
13
  vector_store = Chroma(persist_directory="./bg_data_english")
14
  similarity_retriever = vector_store.as_retriever(
 
9
  from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
10
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, pipeline
11
 
12
+ import os
13
+ from huggingface_hub import login
14
+
15
+ # Authenticate using the token stored in the environment variable
16
+ hf_token = os.getenv("SP_7")
17
+ login(token=hf_token)
18
+
19
+
20
  # Load the pre-existing vector store
21
  vector_store = Chroma(persist_directory="./bg_data_english")
22
  similarity_retriever = vector_store.as_retriever(