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()