Update app.py

app.py

@@ -8,7 +8,14 @@ from nltk.stem import WordNetLemmatizer
 from transformers import pipeline
 from PIL import Image
 
-# Download required NLTK data
+# Set page config FIRST
+device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
+if device_type == "Mobile":
+    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="centered")
+else:
+    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")
+
+# Now the rest of the imports and app logic continue
 nltk.download('stopwords')
 nltk.download('wordnet')
 nltk.download('omw-1.4')
@@ -17,6 +24,7 @@ nltk.download('omw-1.4')
 news_classifier = pipeline("text-classification", model="Oneli/News_Classification")
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
+
 # Label Mapping
 label_mapping = {
     "LABEL_0": "Business",