Spaces:

codeblacks
/

embed

Building

codeblacks commited on Aug 26, 2024

Commit

dbf6cc2

verified ·

1 Parent(s): ff7c210

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,8 @@
 from sentence_transformers import SentenceTransformer
 import gradio as gr
-import torch
 # Load the pre-trained model
-embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
 def get_embeddings(sentences):
     embeddings = model.encode(sentences, convert_to_tensor=True)

 from sentence_transformers import SentenceTransformer
 import gradio as gr
 # Load the pre-trained model
+model = SentenceTransformer('all-MiniLM-L6-v2')
 def get_embeddings(sentences):
     embeddings = model.encode(sentences, convert_to_tensor=True)