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