davidberenstein1957 HF staff commited on
Commit
05e7b4d
·
verified ·
1 Parent(s): 0ca033d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,13 +1,12 @@
 
1
  import gradio as gr
2
-
3
  from sentence_transformers import SentenceTransformer
4
- import duckdb
5
  from huggingface_hub import get_token
6
 
7
- from sentence_transformers import SentenceTransformer
8
- import duckdb
9
 
10
- model = SentenceTransformer("sentence-transformers/static-retrieval-mrl-en-v1")
 
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"