Fix label names

app.py

@@ -2,10 +2,11 @@ import gradio as gr
 from transformers import pipeline
 
 pipe = pipeline("text-classification", model="kitrofimov/news-clf", top_k=3)
+label_names = dict(zip([f"LABEL_{i}" for i in range(4)], ["World", "Sports", "Business", "Sci/Tech"]))
 
 def classify(text):
     preds = pipe(text)[0]
-    return {p["label"]: float(p["score"]) for p in preds}
+    return {label_names[p["label"]]: float(p["score"]) for p in preds}
 
 with gr.Blocks() as demo:
     gr.Markdown("# News Classifier")