Hugging Face's logo

Spaces:
anhkhoiphan
/
ExplrChatbot
Runtime error

App Files Community
anhkhoiphan commited on
Commit
91dcaf2
路
verified 路
1 Parent(s): 79fb0e6

Update chatbot/retrieval.py

Browse files
Files changed (1) hide show
  1. chatbot/retrieval.py +6 -0
chatbot/retrieval.py CHANGED
@@ -6,6 +6,12 @@ import os
6
  import fitz # PyMuPDF
7
  from config import EMBEDDING_MODEL,QDRANT_HOST,QDRANT_API_KEY
8
 
 
 
 
 
 
 
9
  embedding_model = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
10
 
11
  def extract_text_from_pdf(pdf_path):
 
6
  import fitz # PyMuPDF
7
  from config import EMBEDDING_MODEL,QDRANT_HOST,QDRANT_API_KEY
8
 
9
+ EMBEDDING_MODEL = 'keepitreal/vietnamese-sbert'
10
+
11
+ QDRANT_HOST = os.getenv("QDRANT_HOST")
12
+ QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
13
+
14
+
15
  embedding_model = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
16
 
17
  def extract_text_from_pdf(pdf_path):