keithhon's picture
Update app.py
34bdfaf
raw
history blame contribute delete
485 Bytes
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()