Update app.py

app.py

@@ -1,7 +1,7 @@
- import gradio as gr
+import gradio as gr
 from transformers import pipeline
 
-# Load model (Bangla + English supported)
+# Load model
 classifier = pipeline(
     "text-classification",
     model="mrm8488/bert-tiny-finetuned-fake-news-detection"
@@ -15,8 +15,8 @@ def predict_news(text):
     label = result["label"]
     confidence = round(result["score"] * 100, 2)
 
-    # Label mapping
-    if label.upper() == "FAKE":
+    # Correct label mapping
+    if label == "LABEL_0":
         display_label = "Fake"
         color = "red"
     else:
@@ -62,7 +62,83 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             confidence_bar = gr.Slider(
                 minimum=0,
                 maximum=100,
-                label="Accuracy Percentage",
+                label="Confidence Percentage",
+                interactive=False
+            )
+
+    submit_btn.click(
+        fn=predict_news,
+        inputs=news_input,
+        outputs=[prediction_output, confidence_bar]
+    )
+
+demo.launch()
+ 
+import gradio as gr
+from transformers import pipeline
+
+# Load model
+classifier = pipeline(
+    "text-classification",
+    model="mrm8488/bert-tiny-finetuned-fake-news-detection"
+)
+
+def predict_news(text):
+    if text.strip() == "":
+        return "<span style='color:red;'>Please enter news text</span>", 0
+
+    result = classifier(text)[0]
+    label = result["label"]
+    confidence = round(result["score"] * 100, 2)
+
+    # Correct label mapping
+    if label == "LABEL_0":
+        display_label = "Fake"
+        color = "red"
+    else:
+        display_label = "Real"
+        color = "green"
+
+    html_output = f"""
+    <div style="
+        font-size:28px;
+        font-weight:bold;
+        color:{color};
+        text-align:center;
+        padding:20px;
+    ">
+        {display_label} {confidence}%
+    </div>
+    """
+
+    return html_output, confidence
+
+
+# ================= UI LAYOUT =================
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+
+    gr.Markdown(
+        """
+        # 📰 Fake News Detection System  
+        ### AI-based system to detect Fake or Real news (English & Bangla)
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            news_input = gr.Textbox(
+                label="Enter News Text",
+                placeholder="Paste English or Bangla news here...",
+                lines=10
+            )
+            submit_btn = gr.Button("Submit", variant="primary")
+
+        with gr.Column(scale=1):
+            prediction_output = gr.HTML(label="Prediction")
+            confidence_bar = gr.Slider(
+                minimum=0,
+                maximum=100,
+                label="Confidence Percentage",
                 interactive=False
             )