File size: 485 Bytes
b0daf5f
 
a470c1f
b0daf5f
a470c1f
34bdfaf
 
a470c1f
 
 
 
9a9645a
b0daf5f
7c21073
b0daf5f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
from sentence_transformers.util import cos_sim
import numpy as np

def textToFloatVector(text):
    preprocessed = text.replace('[', '').replace(']','');
    x = np.fromstring(preprocessed, dtype=float, sep=',')
    vec = x.astype(np.float)    
    return vec 
    
def compute(text1, text2):
    return  cos_sim(textToFloatVector(text1), textToFloatVector(text2)).tolist()[0]

iface = gr.Interface(fn=compute, inputs=["text","text"], outputs="text")
iface.launch()