Spaces:
Runtime error
Runtime error
Format file
Browse files
perplexity_lenses/engine.py
CHANGED
@@ -12,7 +12,12 @@ from sentence_transformers import SentenceTransformer
|
|
12 |
from perplexity_lenses.visualization import draw_interactive_scatter_plot
|
13 |
|
14 |
logger = logging.getLogger(__name__)
|
15 |
-
EMBEDDING_MODELS = [
|
|
|
|
|
|
|
|
|
|
|
16 |
DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
|
17 |
DOCUMENT_TYPES = ["Whole document", "Sentence"]
|
18 |
SEED = 0
|
|
|
12 |
from perplexity_lenses.visualization import draw_interactive_scatter_plot
|
13 |
|
14 |
logger = logging.getLogger(__name__)
|
15 |
+
EMBEDDING_MODELS = [
|
16 |
+
"distiluse-base-multilingual-cased-v1",
|
17 |
+
"distiluse-base-multilingual-cased-v2",
|
18 |
+
"all-mpnet-base-v2",
|
19 |
+
"flax-sentence-embeddings/all_datasets_v3_mpnet-base",
|
20 |
+
]
|
21 |
DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
|
22 |
DOCUMENT_TYPES = ["Whole document", "Sentence"]
|
23 |
SEED = 0
|