File size: 373 Bytes
eb98c65 ad692df eb98c65 ad692df 8ec69da eb98c65 5918cee 1e15baa 5918cee 6b8b7cd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('sentence-transformers/paraphrase-mpnet-base-v2')
def get_embedding(text):
embedding = model.encode(text)
return embedding.tolist()
iface = gr.Interface(
fn=get_embedding,
inputs=gr.Textbox(),
outputs="json",
api_name="embed"
)
iface.launch()
|