File size: 284 Bytes
72da677 | 1 2 3 4 5 6 7 8 9 10 11 12 | import gradio as gr
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("all-MiniLM-L6-v2")
def embed(text):
embedding = model.encode(text).tolist()
return embedding
iface = gr.Interface(fn=embed, inputs="text", outputs="json")
iface.launch()
|