Update app.py

app.py

@@ -27,6 +27,18 @@ MODELS = {
 USER_PFP = "user.png"  # Hugging Face user avatar
 AI_PFP = "ai_pfp.png"  # Replace with the path to your AI's image or a URL
 
+# Set page config (must be called only once and before any other Streamlit commands)
+st.set_page_config(
+    page_title="Atlas Model Inference",
+    page_icon="🦁 ",
+    layout="wide",
+    menu_items={
+        'Get Help': 'https://huggingface.co/collections/Spestly/athena-1-67623e58bfaadd3c2fcffb86',
+        'Report a bug': 'https://huggingface.co/Spestly/Athena-1-1.5B/discussions/new',
+        'About': 'Athena Model Inference Platform'
+    }
+)
+
 # Custom CSS for blue sliders and button
 st.markdown(
     """
@@ -56,17 +68,6 @@ class AtlasInferenceApp:
         if "chat_history" not in st.session_state:
             st.session_state.chat_history = []
 
-        st.set_page_config(
-            page_title="Atlas Model Inference",
-            page_icon="🦁 ",
-            layout="wide",
-            menu_items={
-                'Get Help': 'https://huggingface.co/collections/Spestly/athena-1-67623e58bfaadd3c2fcffb86',
-                'Report a bug': 'https://huggingface.co/Spestly/Athena-1-1.5B/discussions/new',
-                'About': 'Athena Model Inference Platform'
-            }
-        )
-
     def clear_memory(self):
         """Optimize memory management for CPU inference"""
         if torch.cuda.is_available():