import gradio as gr
from transformers import pipeline
from presentation import main_title, examples

model = 'hackathon-pln-es/Detect-Acoso-Twitter-Es'

article = """
# UNL: Universidad Nacional de Loja
### Miembros del equipo
- Anderson Quizhpe
- Luis Negrón
- David Pacheco
- Bryan Requenes
- Paul Pasaca
"""
def classify(text):
    cls= pipeline("text-classification", model=model)
    return cls(text)[0]['label']

gr.Interface(fn=classify, inputs=[gr.inputs.Textbox(
                lines=5,
                label="Texto a analizar:",
                placeholder="Introduce o elige un ejemplo de twit:",
                optional=False,
            )], outputs=[gr.outputs.Textbox(label="Predicción"),
                              ], description=main_title,
        examples=examples,
        article=article,
        theme="peach",
        thumbnail="None",
        css="https://cdn.jsdelivr.net/npm/bootstrap@3.3.7/dist/css/bootstrap.min.css").launch(debug=True)