Update app.py

app.py

@@ -71,6 +71,8 @@ from peft import PeftModel
 # result = app_module.main()
 
 
+# Set page config as the FIRST command
+st.set_page_config(page_title="📐 Math Assignment Optimizer", layout="wide")
 
 # Helper function to clean up cache
 def clear_cache(cache_dir="private_space_cache"):
@@ -83,9 +85,6 @@ def clear_cache(cache_dir="private_space_cache"):
         except Exception as e:
             st.warning(f"Could not delete {file}: {e}")
 
-# Streamlit app setup
-st.title("Math Question Labeling Tool (Public Interface)")
-
 # Get environment variables
 HF_TOKEN_LLAMA = os.environ.get("HF_TOKEN_LLAMA")
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -112,8 +111,8 @@ with st.spinner("Downloading private space..."):
             repo_type=REPO_TYPE,
             token=HF_TOKEN,
             cache_dir="private_space_cache",
-            force_download=True,  # Always redownload to avoid stale cache
-            local_dir_use_symlinks=False  # Avoid symlinks that might break
+            force_download=True,
+            local_dir_use_symlinks=False
         )
     except Exception as e:
         st.error(f"Failed to download private space: {e}")
@@ -130,7 +129,7 @@ try:
         filename="app.py",
         repo_type=REPO_TYPE,
         token=HF_TOKEN,
-        force_download=True  # Ensure fresh download
+        force_download=True
     )
 except Exception as e:
     st.error(f"Failed to download app.py: {e}")