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()