Update app.py

app.py

@@ -51,7 +51,25 @@ def predict_category(text):
     category_idx = np.argmax(prediction, axis=1)[0]
     return label_encoder.inverse_transform([category_idx])[0]
 
-# Streamlit UI
+# Streamlit UI with Custom Background
+page_bg_img = '''
+<style>
+body {
+    background-image: url("News image 2");
+    background-size: cover;
+    background-position: center;
+    background-repeat: no-repeat;
+}
+.stApp {
+    background: rgba(255, 255, 255, 0.8);
+    padding: 20px;
+    border-radius: 10px;
+}
+</style>
+'''
+
+st.markdown(page_bg_img, unsafe_allow_html=True)
+
 st.title("News Classification App")
 
 # User input