mrolando
first
05b004f
raw
history blame
1.85 kB
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(
"""
<center>
<h1>
Uso de AI para la comparaci贸n de frases y palabras.
</h1>
<img src='data:image/jpg;base64,{}' width=200px>
<h3>
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.
</h3>
</center>
""".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.Column():
with gr.Row():
text_output = gr.Label()
btn.click(process_text, [text_input,text_input2, text_input3,text_input4], text_output)
demo.launch()