Update app.py

app.py

@@ -3,14 +3,21 @@ 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):
-    result = classifier(text)
-    return {res['label']: res['score'] for res in result}
+    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=3),
+    outputs=gr.Label(num_top_classes=4),
     title="Clasificador de Afirmaciones Climáticas",
 )