Spaces:
Sleeping
Sleeping
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() | |