tien314 commited on
Commit
6321de9
·
verified ·
1 Parent(s): 73cdc6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -7,10 +7,15 @@ from langchain.docstore.document import Document
7
  import torch
8
  from langchain_ollama.llms import OllamaLLM
9
  from langchain.chains import LLMChain
 
 
 
10
 
11
 
12
  @st.cache_data
13
  def load_data():
 
 
14
  retriever = BM25HF.load_from_hub(
15
  "tien314/bm25s-version2", load_corpus=True, mmap=True)
16
  return retriever
 
7
  import torch
8
  from langchain_ollama.llms import OllamaLLM
9
  from langchain.chains import LLMChain
10
+ import os
11
+ from huggingface_hub import login
12
+
13
 
14
 
15
  @st.cache_data
16
  def load_data():
17
+
18
+ login(token=os.environ["HF_TOKEN"])
19
  retriever = BM25HF.load_from_hub(
20
  "tien314/bm25s-version2", load_corpus=True, mmap=True)
21
  return retriever