Update app.py

app.py

@@ -31,12 +31,13 @@ def preprocess_text(text):
     return ' '.join(tokens)
 
 
-classifier_model = "Ginidu2003/Distilbert-Base-News-classifier"
+
 
 
 
 
 # ====================== CLASSIFICATION FUNCTION ======================
+classifier_model = "Ginidu2003/Distilbert-Base-News-classifier"
 @torch.no_grad()
 def classify_csv(file):
     try:
@@ -95,7 +96,7 @@ def create_colored_bar_chart(category_counts):
         ax.text(bar.get_x() + bar.get_width()/2, height + 0.5, 
                 str(int(height)), ha='center', va='bottom', fontsize=12, fontweight='bold')
 
-    ax.set_title("Category Distribution Across 5 Classes", fontsize=14, fontweight='bold')
+    ax.set_title("News Category Distribution Across 5 Classes", fontsize=14, fontweight='bold')
     ax.set_xlabel("Category")
     ax.set_ylabel("Count")
     plt.xticks(rotation=15)
@@ -135,9 +136,9 @@ def answer_question(news_content, question):
         return f"Error: {str(e)}"
 
 # ====================== GRADIO INTERFACE ======================
-with gr.Blocks(title=" News Classifier & Question Answering App") as demo:
-    gr.Markdown("# 📰 English News Classifier")
-    #gr.Markdown("### Section 02 - Text Analytics Assignment")
+with gr.Blocks(title="News Classifier & Question Answering App..") as demo:
+    gr.Markdown("# 📰 News Classifier & Question Answering App..")
+    
 
     with gr.Tabs():
         with gr.Tab("📊 News Classification"):
@@ -147,7 +148,7 @@ with gr.Blocks(title=" News Classifier & Question Answering App") as demo:
             output_text = gr.Textbox(label="Status")
             output_file = gr.File(label="Download output.csv")
             bar_chart = gr.Plot(
-                 label="Category Distribution Across 5 Classes"
+                 label="News Category Distribution Across 5 Classes"
                 
                  
             )
@@ -171,6 +172,6 @@ with gr.Blocks(title=" News Classifier & Question Answering App") as demo:
                 outputs=qa_output
             )
 
-    #gr.Markdown("Built for Text Analytics Assignment - Section 02")
+    
 
 demo.launch()