keithhon's picture
Update app.py
a470c1f
raw
history blame
392 Bytes
import gradio as gr
from sentence_transformers.util import cos_sim
import numpy as np
def textToFloatVector(text):
x = np.array(text1.split(","))
vec = x.astype(np.float)
return vec
def compute(text1, text2):
return cos_sim(textToFloatVector(text1), textToFloatVector(text2))
iface = gr.Interface(fn=compute, inputs=["text","text"], outputs="text")
iface.launch()