Update app.py

app.py

@@ -2,14 +2,12 @@ import gradio as gr
 from transformers import pipeline
 
 # Load zero-shot classifier for bias detection using Facebook's BART MNLI.
-# This model will classify the input text into "biased" or "neutral".
 classifier = pipeline(
     "zero-shot-classification",
     model="facebook/bart-large-mnli"
 )
 
 # Load FLAN-T5 for rewriting text.
-# This model will generate a neutral version of biased text.
 rewriter = pipeline(
     "text2text-generation",
     model="google/flan-t5-base",
@@ -38,7 +36,7 @@ def process_text(text):
     # Return results in a dictionary.
     return {
         "Detected Bias": detected_bias,
-        "Confidence": confidence,
+        "Confidence": round(confidence, 2),
         "Rewritten Text": rewritten_text
     }
 
@@ -56,8 +54,10 @@ with gr.Blocks() as demo:
         placeholder="Type a story or sentence here...",
         lines=5
     )
+    
+    submit_btn = gr.Button("Submit")
     result_output = gr.JSON(label="Output")
     
-    text_input.submit(process_text, inputs=[text_input], outputs=[result_output])
+    submit_btn.click(fn=process_text, inputs=[text_input], outputs=[result_output])
     
 demo.launch()