Spaces:
Runtime error
Runtime error
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() |