File size: 572 Bytes
475ba97
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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()