apitest1 / app.py
GoldDany's picture
Update app.py
25d16ac verified
raw
history blame
305 Bytes
import gradio as gr
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("all-mpnet-base-v2")
def get_embeddings(text):
return model.encode(text).tolist()
gr.Interface(
fn=get_embedding,
inputs="text",
outputs="json",
title="DiDef Embeddings"
).launch()