Spaces:

ai-blueprint
/

rag-retrieve

Running

davidberenstein1957 commited on Jan 15

Commit

6baafff

verified ·

1 Parent(s): 0636ee3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from sentence_transformers.models import StaticEmbedding
 import duckdb
 # Initialize a StaticEmbedding module
-static_embedding = StaticEmbedding.from_model2vec("minishlab/potion-base-8M")
 model = SentenceTransformer(modules=[static_embedding])
 dataset_name = "smol-blueprint/fineweb-bbc-news-text-embeddings"

 import duckdb
 # Initialize a StaticEmbedding module
+static_embedding = StaticEmbedding.from_model2vec("sentence-transformers/static-retrieval-mrl-en-v1")
 model = SentenceTransformer(modules=[static_embedding])
 dataset_name = "smol-blueprint/fineweb-bbc-news-text-embeddings"