Pudding48 commited on
Commit
eb6eee8
·
verified ·
1 Parent(s): 52ac670

Update qabot.py

Browse files
Files changed (1) hide show
  1. qabot.py +4 -2
qabot.py CHANGED
@@ -29,9 +29,11 @@ model_file = hf_hub_download(
29
  # Vector store location
30
  vector_dp_path = "/home/user/app/vectorstores/db_faiss"
31
 
 
 
32
  from prepare_vector_dp import create_dp_from_files
33
  #create_db_from_text()
34
- create_dp_from_files()
35
 
36
  # Load LLM with CTransformers
37
  def load_llm(model_file):
@@ -60,7 +62,7 @@ def create_qa_chain(prompt, llm, db):
60
 
61
  # Load vector DB
62
  def read_vector_db():
63
- embedding_model = HuggingFaceEmbeddings(model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
64
  return FAISS.load_local(vector_dp_path, embedding_model, allow_dangerous_deserialization=True)
65
 
66
  # Build everything
 
29
  # Vector store location
30
  vector_dp_path = "/home/user/app/vectorstores/db_faiss"
31
 
32
+ request_model = "sentence-transformers/distiluse-base-multilingual-cased-v1"
33
+
34
  from prepare_vector_dp import create_dp_from_files
35
  #create_db_from_text()
36
+ create_dp_from_files(request_model)
37
 
38
  # Load LLM with CTransformers
39
  def load_llm(model_file):
 
62
 
63
  # Load vector DB
64
  def read_vector_db():
65
+ embedding_model = HuggingFaceEmbeddings(model_name = request_model)
66
  return FAISS.load_local(vector_dp_path, embedding_model, allow_dangerous_deserialization=True)
67
 
68
  # Build everything