Trabis commited on
Commit
e42d0b9
·
verified ·
1 Parent(s): a5308de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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 ffff"}
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