File size: 334 Bytes
e46be7c | 1 2 3 4 5 6 7 8 9 10 11 | import gradio as gr
import json
from sentence_transformers import SentenceTransformer
model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
def give_embeddings(sentence):
return json.dumps(model.encode([sentence])[0].tolist())
demo = gr.Interface(fn=give_embeddings, inputs="text", outputs="text")
demo.launch() |