Spaces:
Runtime error
Runtime error
import gradio as gr | |
import examples | |
from document_quieries import ExtractiveProposalQueries | |
from config import Config | |
query = ExtractiveProposalQueries(es_host = Config.es_host, es_index = Config.proposals_index, | |
es_user = Config.es_user, es_password = Config.es_password, | |
reader_name_or_path = Config.reader_model_name_or_path, | |
use_gpu = Config.use_gpu) | |
def search(question): | |
p1_result = query.search_by_query(query = question, retriever_top_k = 2, reader_top_k = 1, es_index = "petro") | |
p2_result = query.search_by_query(query = question, retriever_top_k = 2, reader_top_k = 1, es_index = "rodolfo") | |
return [p1_result[0].answer, | |
p1_result[0].context, | |
p2_result[0].answer, | |
p2_result[0].context] | |
description = """ | |
<p> | |
El debate ha sido un ausente en la segunda vuelta por la presidencia de Colombia del 2022. | |
para ayudar con el conocimiento de las propuestas de los candidatos construí este espacio, | |
puede hacerle preguntas a la propuestas oficiales de los candidatos (Junio de 2022). | |
Un pequeño aporte a un fin de semana determinante en la democracia Colombiana. | |
<br></br> | |
<br>Este espacio hace parte de un proyecto open source que busca ayudar con el entendimiento de temas relevantes para el país, como estas elecciones y la memoria histórica Colombiana. <a href= "https://github.com/jorge-henao/ask_to_democracy">repo en github con FastAPI</a></br> | |
Por: Jorge Henao 🇨🇴 <a href="https://twitter.com/jhenaotw" target='_blank'>Twitter</a> <a href="https://www.linkedin.com/in/henaojorge" target='_blank'/>LinkedIn</a> | |
</p> | |
""" | |
iface = gr.Interface(fn=search, | |
title= "Pregúntale a la democracia 🇨🇴 - Elecciones precidenciales Colombia 2022", | |
description = description, | |
inputs=[ | |
gr.inputs.Textbox(lines=2, label="hazle una pregunta a los candidatos [Abajo algunos ejemplos]👇", placeholder='que va a hacer con la deuda del ICETEX?') | |
], | |
outputs=[ | |
gr.outputs.Textbox(label="Respuesta de Petro"), | |
gr.outputs.Textbox(label="Contexto de la propuesta"), | |
gr.outputs.Textbox(label="Respuesta de Rodolfo Hernandez"), | |
gr.outputs.Textbox(label="Contexto de la propuesta") | |
], | |
examples = examples.examples, | |
theme = 'peach' | |
) | |
iface.launch(enable_queue=True) |