Spaces:
Sleeping
Sleeping
Commit
·
703587b
1
Parent(s):
7f19939
chunk siz = 1000, max_chars = 1500
Browse files
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="
|
| 11 |
-
"""
|
| 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"""
|