Update app.py
Browse files
app.py
CHANGED
@@ -399,7 +399,7 @@ def get_text_splitter(split_strategy, chunk_size, overlap_size, custom_separator
|
|
399 |
def get_embedding_model(model_type, model_name):
|
400 |
model_path = model_manager.get_model(model_type, model_name)
|
401 |
if model_type == 'HuggingFace':
|
402 |
-
return
|
403 |
model_name=model_path,
|
404 |
multi_process=True,
|
405 |
model_kwargs={"device": "cuda"},
|
|
|
399 |
def get_embedding_model(model_type, model_name):
|
400 |
model_path = model_manager.get_model(model_type, model_name)
|
401 |
if model_type == 'HuggingFace':
|
402 |
+
return HuggingFaceEmbeddings(
|
403 |
model_name=model_path,
|
404 |
multi_process=True,
|
405 |
model_kwargs={"device": "cuda"},
|