Update label definitions in app.py to include label_id mapping for improved clarity and organization.

app.py

@@ -2,9 +2,20 @@ import gradio as gr
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
-# Định nghĩa các nhãn
-LABELS = ['Thế giới', 'Văn hóa', 'Chính trị Xã hội', 'Vi tính', 'Đời sống', 
-          'Thể thao', 'Sức khỏe', 'Kinh doanh', 'Pháp luật', 'Khoa học']
+# Định nghĩa các nhãn theo label_id (0-9)
+# Mapping: label_id -> label
+LABELS = [
+    'Chính trị Xã hội', # label_id 0
+    'Khoa học',         # label_id 1
+    'Kinh doanh',       # label_id 2
+    'Pháp luật',        # label_id 3
+    'Sức khỏe',         # label_id 4
+    'Thế giới',         # label_id 5
+    'Thể thao',         # label_id 6
+    'Vi tính',          # label_id 7
+    'Văn hóa',          # label_id 8
+    'Đời sống',         # label_id 9
+]
 
 # Khởi tạo device
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')