Update app.py

app.py

@@ -11,7 +11,7 @@ model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
 # ===== Hàm dự đoán có progress bar =====
 def predict_with_progress(text, progress=gr.Progress(track_tqdm=True)):
     if not text.strip():
-        return "❌ Vui lòng nhập bình luận", 0.0, "gray"
+        return "❌ Vui lòng nhập bình luận", 0.0
 
     progress(0.1, desc="Đang xử lý văn bản...")
     time.sleep(0.3)
@@ -23,14 +23,13 @@ def predict_with_progress(text, progress=gr.Progress(track_tqdm=True)):
         outputs = model(**inputs)
         probs = torch.softmax(outputs.logits, dim=1).tolist()[0]
 
-    label = "Toxic" if probs[1] > probs[0] else "Non-toxic"
+    label = "🧨 Toxic" if probs[1] > probs[0] else "✅ Non-toxic"
     confidence = max(probs)
 
     progress(0.9, desc="Hoàn tất!")
     time.sleep(0.2)
 
-    color = "red" if label == "Toxic" else "green"
-    return f"🔹 **Kết quả:** {label}", confidence, color
+    return f"**Kết quả:** {label}", confidence
 
 
 # ===== Giao diện Gradio =====
@@ -38,32 +37,27 @@ with gr.Blocks(title="🛡️ Toxic Comment Detector") as demo:
     gr.Markdown("## 🧠 Phân loại bình luận độc hại (Toxic Comment Detector)")
     gr.Markdown(
         "Nhập một đoạn bình luận bằng tiếng Việt để kiểm tra mức độ độc hại.\n\n"
-        "Mô hình sử dụng: `vijjj1/toxic-comment-phobert`"
+        "Mô hình: **vijjj1/toxic-comment-phobert**"
     )
 
-    with gr.Row():
-        txt_input = gr.Textbox(
-            label="Nhập bình luận cần kiểm tra",
-            placeholder="Ví dụ: 'Mày ngu như bò vậy!'",
-            lines=3
-        )
+    txt_input = gr.Textbox(
+        label="Nhập bình luận cần kiểm tra",
+        placeholder="Ví dụ: 'Mày ngu như bò vậy!'",
+        lines=3
+    )
 
     btn = gr.Button("🚀 Phân tích bình luận", variant="primary")
-    output_label = gr.Markdown(label="Kết quả")
-    prob_bar = gr.Progress(label="Độ tin cậy", value=0)
 
-    with gr.Row():
-        conf_bar = gr.Slider(
-            minimum=0, maximum=1, value=0, step=0.01, label="Độ tin cậy (Confidence)", interactive=False
-        )
+    output_label = gr.Markdown(label="Kết quả")
+    conf_bar = gr.Slider(
+        minimum=0, maximum=1, value=0, step=0.01, label="Độ tin cậy (Confidence)", interactive=False
+    )
 
-    # Khi bấm nút
     btn.click(
         fn=predict_with_progress,
-        inputs=[txt_input],
-        outputs=[output_label, conf_bar, None]
+        inputs=txt_input,
+        outputs=[output_label, conf_bar]
     )
 
-# Chạy local hoặc deploy HF Spaces
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)