Update app.py

app.py

@@ -8,20 +8,22 @@ from nltk.stem import WordNetLemmatizer
 from transformers import pipeline
 from PIL import Image
 
-# Check if device type is already stored
+# Ensure set_page_config is the first Streamlit command
 if "device_type" not in st.session_state:
+    st.set_page_config(page_title="News Classifier", page_icon="📰", layout="wide")  # Default layout
+
     # Ask user for device type
     device_type = st.radio("Are you using a mobile device or a PC?", ["Mobile", "PC"])
     st.session_state.device_type = device_type
     st.experimental_rerun()  # Refresh the app to apply layout change
 
-# Set layout based on user selection
+# Apply the selected layout before anything else
 if st.session_state.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")
 
-# Continue with the rest of the code
+# Now continue with the rest of your code
 nltk.download('stopwords')
 nltk.download('wordnet')
 nltk.download('omw-1.4')