Update app.py

app.py

@@ -35,7 +35,7 @@ lemmatizer = WordNetLemmatizer()
 # Load models (cache them to avoid reloading on every interaction)
 @st.cache_resource
 def load_classification_model():
-    model_name = "Prageeth-1/News_classification.2"  # Replace with your model path
+    model_name = "Imasha17/News_classification.3"  # Replace with your model path
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return pipeline("text-classification", model=model, tokenizer=tokenizer)
@@ -86,14 +86,14 @@ st.markdown("""
         display: flex;
         align-items: center;
         margin-bottom: 20px;
-        colour : #2196F3
+        background-color: #2196F3;
     }
     .header img {
         height: 50px;  
         margin-right: 10px;
     }
     .header h1 {
-        font-size: 24px;
+        font-size: 40px;
         color: #4CAF50;  
         margin: 0;