Update app.py

app.py

@@ -7,16 +7,7 @@ def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'
 device = get_device_map()  # 'cpu'
 
-def setup_page():
-    with st.spinner("In progress..."):
-        hf_token = os.getenv('HF_TOKEN')
-        df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv") 
-        st.image("images/SBS_Chapter_Index.png", use_container_width=True) 
-        st.subheader("Select specific Chapter for quicker results")
-        st.write(df_chapters.head())
-
-
-    custom_css = f"""
+custom_css = f"""
     <style>
         /* Making the Streamlit header sticky */
         {HEADER_SELECTOR} {{
@@ -32,6 +23,15 @@ def setup_page():
             /* }} */
         }}
 
+
+def setup_page():
+    with st.spinner("In progress..."):
+        hf_token = os.getenv('HF_TOKEN')
+        df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv") 
+        st.image("images/SBS_Chapter_Index.png", use_container_width=True) 
+        st.subheader("Select specific Chapter for quicker results")
+        st.write(df_chapters.head())    
+
         /* Adding padding to the main content area to prevent overlap with the sticky header */
         {MAIN_CONTENT_SELECTOR} {{
             padding-top: 10px !important; /* {HEADER_HEIGHT} !important; */