Update src/models/plot_similarity_finder.py
Browse filesCambié el modelo a model_name="sentence-t5-xl" (antes era model_name="sentence-t5-large").
src/models/plot_similarity_finder.py
CHANGED
@@ -6,7 +6,7 @@ from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddi
|
|
6 |
|
7 |
# Searching for similar plots to the users plot/input
|
8 |
# Loading embedding function
|
9 |
-
embedding_function = SentenceTransformerEmbeddings(model_name="sentence-t5-
|
10 |
|
11 |
# Loading vectorstore
|
12 |
db = FAISS.load_local('./streamlit/plot_embeddings', embedding_function)
|
|
|
6 |
|
7 |
# Searching for similar plots to the users plot/input
|
8 |
# Loading embedding function
|
9 |
+
embedding_function = SentenceTransformerEmbeddings(model_name="sentence-t5-xl")
|
10 |
|
11 |
# Loading vectorstore
|
12 |
db = FAISS.load_local('./streamlit/plot_embeddings', embedding_function)
|