enoreyes commited on
Commit
9f0e33c
1 Parent(s): d0e60eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,8 @@ from langchain.llms import HuggingFaceHub
8
 
9
  from chain import get_new_chain1
10
 
 
 
11
  def get_faiss_store():
12
  with open("docs.pkl", 'rb') as f:
13
  faiss_store = pickle.load(f)
@@ -22,7 +24,7 @@ def load_model():
22
 
23
  flan_ul = HuggingFaceHub(repo_id="google/flan-ul2",
24
  model_kwargs={"temperature":0.1, "max_new_tokens":200},
25
- huggingfacehub_api_token="")
26
 
27
  qa_chain = get_new_chain1(vectorstore, flan_ul, flan_ul, isFlan=True)
28
  return qa_chain
 
8
 
9
  from chain import get_new_chain1
10
 
11
+ api_token = os.environ["HF_TOKEN"]
12
+
13
  def get_faiss_store():
14
  with open("docs.pkl", 'rb') as f:
15
  faiss_store = pickle.load(f)
 
24
 
25
  flan_ul = HuggingFaceHub(repo_id="google/flan-ul2",
26
  model_kwargs={"temperature":0.1, "max_new_tokens":200},
27
+ huggingfacehub_api_token=api_token)
28
 
29
  qa_chain = get_new_chain1(vectorstore, flan_ul, flan_ul, isFlan=True)
30
  return qa_chain