MrSimple07 commited on
Commit
703587b
·
1 Parent(s): 7f19939

chunk siz = 1000, max_chars = 1500

Browse files
Files changed (1) hide show
  1. utils.py +3 -6
utils.py CHANGED
@@ -7,12 +7,9 @@ def get_llm_model(api_key, model_name="gemini-2.0-flash"):
7
  """Get LLM model"""
8
  return GoogleGenAI(model=model_name, api_key=api_key)
9
 
10
- def get_embedding_model(model_name="intfloat/multilingual-e5-large"):
11
- """Use better multilingual embedding model"""
12
- return HuggingFaceEmbedding(
13
- model_name=model_name,
14
- trust_remote_code=True
15
- )
16
 
17
  def get_reranker_model(model_name='cross-encoder/ms-marco-MiniLM-L-12-v2'):
18
  """Get reranker model"""
 
7
  """Get LLM model"""
8
  return GoogleGenAI(model=model_name, api_key=api_key)
9
 
10
+ def get_embedding_model(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"):
11
+ """Get embedding model"""
12
+ return HuggingFaceEmbedding(model_name=model_name)
 
 
 
13
 
14
  def get_reranker_model(model_name='cross-encoder/ms-marco-MiniLM-L-12-v2'):
15
  """Get reranker model"""