Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
import gradio as gr
|
2 |
-
from
|
3 |
|
4 |
# Load the Nomic embedding model
|
5 |
-
model =
|
6 |
|
7 |
def get_embedding(text):
|
8 |
"""Generate an embedding for the input text using Nomic encoder."""
|
@@ -10,7 +10,7 @@ def get_embedding(text):
|
|
10 |
return "Please provide some text."
|
11 |
|
12 |
# Generate embedding
|
13 |
-
embedding = model.
|
14 |
|
15 |
# Return embedding as list (more user-friendly in the UI)
|
16 |
return embedding.tolist()
|
|
|
1 |
import gradio as gr
|
2 |
+
from sentence_transformers import SentenceTransformer
|
3 |
|
4 |
# Load the Nomic embedding model
|
5 |
+
model = SentenceTransformer("nomic-ai/nomic-embed-text-v1", trust_remote_code=True)
|
6 |
|
7 |
def get_embedding(text):
|
8 |
"""Generate an embedding for the input text using Nomic encoder."""
|
|
|
10 |
return "Please provide some text."
|
11 |
|
12 |
# Generate embedding
|
13 |
+
embedding = model.encode([text])[0] # Get the first (and only) embedding
|
14 |
|
15 |
# Return embedding as list (more user-friendly in the UI)
|
16 |
return embedding.tolist()
|