Spaces:
Running
Running
Update app.py
Browse files
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 |
|