HConley commited on
Commit
fda2c79
·
1 Parent(s): 9b28bff

Update src/models/plot_similarity_finder.py

Browse files

Cambié 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-large")
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)