AreejMehboob commited on
Commit
97a4dbc
·
verified ·
1 Parent(s): 6c1e871

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
- from nomic.embed import EmbeddingModel
3
 
4
  # Load the Nomic embedding model
5
- model = EmbeddingModel()
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.embed(text)
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()