Update app.py

app.py

@@ -4,10 +4,10 @@ from transformers import pipeline, AutoConfig
 MODEL_NAME = "blaikhole/distilbert-review-bug-classifier"
 
 label_mapping = {
-    "LABEL_0": "Graphics 🎨",
-    "LABEL_1": "Network 🌐",
+    "LABEL_0": "Graphics issue 🎨",
+    "LABEL_1": "Network issue 🌐",
     "LABEL_2": "No Bug ✅",
-    "LABEL_3": "Performance 🚀"
+    "LABEL_3": "Performance issue 🚀"
 }
 
 # Load model config to get label mapping
@@ -31,7 +31,7 @@ if st.button("Classify"):
 
         predictions = {label_mapping.get(res['label'], res['label']): int(res['score'] * 100) for res in result}
         
-        ordered_labels = ["Graphics 🎨", "Network 🌐", "No Bug ✅", "Performance 🚀"]
+        ordered_labels = ["Graphics issue 🎨", "Network issue 🌐", "No Bug ✅", "Performance issue 🚀"]
         ordered_predictions = {k: predictions[k] for k in ordered_labels if k in predictions}
 
         # Get top prediction