File size: 428 Bytes
ea71e3a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import gradio as gr
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("jinaai/jina-embeddings-v2-small-en")

def embed(text):
    embedding = model.encode(text).tolist()
    return embedding

gr.Interface(
    fn=embed,
    inputs=gr.Textbox(label="Text"),
    outputs="json",
    title="Jina Embedding Model",
    description="Get embeddings using jinaai/jina-embeddings-v2-small-en"
).launch()