kanninian commited on
Commit
becf13f
·
verified ·
1 Parent(s): 280414d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from qa_vector_store import build_qa_vector_store, retrieve_and_rerank, generate
13
  # 初始化模型和資料庫
14
  model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
15
  collection_name = model_name.split("/")[-1]
16
- cross_encoder_model = "cross-encoder/mmarco-mMiniLMv2-L12-H384-v1"
17
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
18
  llm_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
19
 
 
13
  # 初始化模型和資料庫
14
  model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
15
  collection_name = model_name.split("/")[-1]
16
+ cross_encoder_model = CrossEncoder("cross-encoder/mmarco-mMiniLMv2-L12-H384-v1")
17
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
18
  llm_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
19