Update app.py
Browse files
app.py
CHANGED
@@ -78,7 +78,7 @@ bm25 = BM25Encoder().load("./bm25_uae_legislation_data.json")
|
|
78 |
# old_embed_model = HuggingFaceEmbeddings(model_name="sentence-transformers/gte-multilingual-base")
|
79 |
|
80 |
# Initialize models and retriever
|
81 |
-
embed_model = HuggingFaceEmbeddings(model_name="Alibaba-NLP/gte-
|
82 |
retriever = PineconeHybridSearchRetriever(
|
83 |
embeddings=embed_model,
|
84 |
sparse_encoder=bm25,
|
|
|
78 |
# old_embed_model = HuggingFaceEmbeddings(model_name="sentence-transformers/gte-multilingual-base")
|
79 |
|
80 |
# Initialize models and retriever
|
81 |
+
embed_model = HuggingFaceEmbeddings(model_name="Alibaba-NLP/gte-large-en-v1.5", model_kwargs={"trust_remote_code":True})
|
82 |
retriever = PineconeHybridSearchRetriever(
|
83 |
embeddings=embed_model,
|
84 |
sparse_encoder=bm25,
|