rodrigomoreirasilva
/

bertopic_lai_recursos_CGU

Model card Files Files and versions Community

rodrigomoreirasilva commited on Dec 7, 2023

Commit

fd10c55

·

1 Parent(s): a2cb5db

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension
 embedding_model = SentenceTransformer(modules=[word_embedding_model, pooling_model])
 topic_model = BERTopic.load("rodrigomoreirasilva/bertopic_lai_recursos_CGU",
-                            embedding_model='efederici/e5-base-multilingual-4096')
 topic_model.get_topic_info()
 #Função para tratar o texto antes do calculo das embeddings

 embedding_model = SentenceTransformer(modules=[word_embedding_model, pooling_model])
 topic_model = BERTopic.load("rodrigomoreirasilva/bertopic_lai_recursos_CGU",
+                            embedding_model=embedding_model)
 topic_model.get_topic_info()
 #Função para tratar o texto antes do calculo das embeddings