Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,11 @@ import re
|
|
16 |
|
17 |
class DocumentRetrievalAndGeneration:
|
18 |
def __init__(self, embedding_model_name, lm_model_id, data_folder):
|
19 |
-
hf_token = os.getenv('HF_TOKEN')
|
20 |
-
|
|
|
|
|
|
|
21 |
self.all_splits = self.load_documents(data_folder)
|
22 |
self.embeddings = SentenceTransformer(embedding_model_name)
|
23 |
self.cpu_index = self.create_faiss_index()
|
|
|
16 |
|
17 |
class DocumentRetrievalAndGeneration:
|
18 |
def __init__(self, embedding_model_name, lm_model_id, data_folder):
|
19 |
+
# hf_token = os.getenv('HF_TOKEN')
|
20 |
+
hf="hf_VuNNBwnFqlcKzV"
|
21 |
+
token="vCfLXEBxyAOftxvlWpwf"
|
22 |
+
hf_token=hf+token
|
23 |
+
# print(HF_TOKEN,hf_token)
|
24 |
self.all_splits = self.load_documents(data_folder)
|
25 |
self.embeddings = SentenceTransformer(embedding_model_name)
|
26 |
self.cpu_index = self.create_faiss_index()
|