from sentence_transformers import SentenceTransformer, util import gradio as gr import base64 model = SentenceTransformer('hiiamsid/sentence_similarity_spanish_es') def process_text(text1,text2,text3,text4): embeddings = model.encode([text1,text2,text3,text4],convert_to_tensor=True) score = util.cos_sim(embeddings,embeddings).numpy() return {text2 : score[0][1].item(),text3 : score[0][2].item(),text4 : score[0][3].item()} demo = gr.Blocks(title="Hola") def complete_with_gpt(text1,text2,text3): return text1+text2+text3 with open("Iso_Logotipo_Ceibal.png", "rb") as image_file: encoded_image = base64.b64encode(image_file.read()).decode() with demo: gr.Markdown( """

Uso de AI para la comparación de frases y palabras.

Con este espacio podrás comparar la similitud entre distintas frases y palabras. El resultado es el porcentaje de similitud que el modelo encuentra entre la primer frase en comparación con las siguientes.

""".format(encoded_image)) with gr.Row(): with gr.Column(): with gr.Row(): gr.Markdown("Primero ingresá la primer frase:") with gr.Row(): text_input = gr.Textbox( label="") with gr.Row(): gr.Markdown("Ahora ingresá las frases a comparar con la primera:") with gr.Row(): text_input2 = gr.Textbox(label="Frase 1") with gr.Row(): text_input3 = gr.Textbox(label="Frase 2") with gr.Row(): text_input4 = gr.Textbox(label="Frase 3") with gr.Row(): btn = gr.Button("Calcular") with gr.Row(): examples = gr.Examples([["Hola, ¿Cómo estás?","Hola, ¿Todo bien?","Que feo está el día","Yo estoy bien"]],[text_input,text_input2, text_input3,text_input4]) with gr.Column(): with gr.Row(): text_output = gr.Label() btn.click(process_text, [text_input,text_input2, text_input3,text_input4], text_output) demo.launch()