File size: 544 Bytes
1d8387c
d3f68b0
 
1d8387c
 
 
 
30857d9
e3805eb
 
e9a9607
e3805eb
 
 
e583f80
1d8387c
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from sentence_transformers import SentenceTransformer
import gradio as gr

model = SentenceTransformer('sentence-transformers/multi-qa-mpnet-base-cos-v1')


def generate_embedding(query): 
    # return "Connected"
    embeddings = model.encode(query)
    resposne = ""
    for char in embeddings.tolist():
        resposne += str(char) + " "
    # mySeparator = " "
    # resposne = mySeparator.join(embeddings.tolist())
    return resposne

demo = gr.Interface(fn=generate_embedding, inputs="text", outputs="text")

demo.launch( show_api=True)