TheTharindu's picture
Add initial implementation of text embedding with Gradio interface
72da677
Raw
History Blame Contribute Delete
284 Bytes
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()