File size: 334 Bytes
5da5b63 0cbead6 5da5b63 f84e003 |
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() |