Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +1,12 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
-
|
3 |
from sentence_transformers import SentenceTransformer
|
4 |
-
import
|
5 |
from huggingface_hub import get_token
|
6 |
|
7 |
-
from sentence_transformers import SentenceTransformer
|
8 |
-
import duckdb
|
9 |
|
10 |
-
|
|
|
11 |
dataset_name = "smol-blueprint/fineweb-bbc-news-text-embeddings"
|
12 |
embedding_column = "embedding"
|
13 |
table_name = "fineweb"
|
|
|
1 |
+
import duckdb
|
2 |
import gradio as gr
|
|
|
3 |
from sentence_transformers import SentenceTransformer
|
4 |
+
from sentence_transformers.models import StaticEmbedding
|
5 |
from huggingface_hub import get_token
|
6 |
|
|
|
|
|
7 |
|
8 |
+
static_embedding = StaticEmbedding.from_model2vec("minishlab/potion-base-8M")
|
9 |
+
model = SentenceTransformer(modules=[static_embedding])
|
10 |
dataset_name = "smol-blueprint/fineweb-bbc-news-text-embeddings"
|
11 |
embedding_column = "embedding"
|
12 |
table_name = "fineweb"
|