Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,8 @@ from sentence_transformers import SentenceTransformer
|
|
12 |
import pickle
|
13 |
import re
|
14 |
|
|
|
|
|
15 |
class RAGLoader:
|
16 |
def __init__(self,
|
17 |
docs_folder: str = "./docs",
|
@@ -47,8 +49,9 @@ class RAGLoader:
|
|
47 |
self.indexed_documents = None
|
48 |
|
49 |
def encode(self,payload):
|
|
|
50 |
API_URL = "https://api-inference.huggingface.co/models/intfloat/multilingual-e5-large"
|
51 |
-
headers = {"Authorization": "Bearer
|
52 |
response = requests.post(API_URL, headers=headers, json=payload)
|
53 |
return response.json()
|
54 |
|
|
|
12 |
import pickle
|
13 |
import re
|
14 |
|
15 |
+
os.environ.get('HUGGINGFACE_TOKEN')
|
16 |
+
|
17 |
class RAGLoader:
|
18 |
def __init__(self,
|
19 |
docs_folder: str = "./docs",
|
|
|
49 |
self.indexed_documents = None
|
50 |
|
51 |
def encode(self,payload):
|
52 |
+
token = os.environ.get('HUGGINGFACE_TOKEN')
|
53 |
API_URL = "https://api-inference.huggingface.co/models/intfloat/multilingual-e5-large"
|
54 |
+
headers = {"Authorization": "Bearer {token}"}
|
55 |
response = requests.post(API_URL, headers=headers, json=payload)
|
56 |
return response.json()
|
57 |
|