Ritesh-hf commited on
Commit
502ee1c
1 Parent(s): fee75d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-multilingual-base", model_kwargs={"trust_remote_code":True})
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,