marekm4's picture
Semantic similarity
475ba97
raw
history blame contribute delete
572 Bytes
import gradio as gr
from simple_nlp_library import embeddings, preprocessing, metrics
vectors = embeddings.vectors()
stop_words = preprocessing.stop_words()
def semantic_similarity(first_sentence, second_sentence):
return metrics.cosine_similarity(
embeddings.tokens_vector(vectors, preprocessing.semantic_tokens(stop_words, first_sentence)),
embeddings.tokens_vector(vectors, preprocessing.semantic_tokens(stop_words, second_sentence)),
)
gr.Interface(
semantic_similarity,
inputs=["text", "text"],
outputs=["number"],
).launch()