File size: 745 Bytes
265a794 9a22885 265a794 9a22885 265a794 9a22885 265a794 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from transformers import pipeline
import gradio as gr
classifier = pipeline("text-classification", model="joortif/clasificador-climate-claim")
label_map = {
"LABEL_0": "Afirma el cambio climático (SUPPORTS)",
"LABEL_1": "Niega el cambio climático (REFUTES)",
"LABEL_2": "No hay suficiente información (NOT_ENOUGH_INFO)",
"LABEL_3": "Afirmación disputada (DISPUTED)"
}
def classify_text(text):
results = classifier(text)
return {label_map[res["label"]]: res["score"] for res in results}
demo = gr.Interface(
fn=classify_text,
inputs=gr.Textbox(lines=4, label="Escribe un texto sobre el clima"),
outputs=gr.Label(num_top_classes=4),
title="Clasificador de Afirmaciones Climáticas",
)
demo.launch() |