Update Gallery.py

pages/Gallery.py

@@ -422,25 +422,27 @@ def load_hf_dataset():
     return roster, promptBook, images_ds
 
 
-if __name__ == "__main__":
-    st.set_page_config(page_title="Model Coffer Gallery", page_icon="🖼️", layout="wide")
-
-    # remove ranking in the session state if it is created in Ranking.py
-    st.session_state.pop('ranking', None)
-
-    if 'user_id' not in st.session_state:
-        st.warning('Please log in first.')
-        home_btn = st.button('Go to Home Page')
-        if home_btn:
-            switch_page("home")
-    else:
-        st.write('You have already logged in as ' + st.session_state.user_id[0])
-        roster, promptBook, images_ds = load_hf_dataset()
-        # print(promptBook.columns)
-
-        # initialize selected_dict
-        if 'selected_dict' not in st.session_state:
-            st.session_state['selected_dict'] = {}
-
-        app = GalleryApp(promptBook=promptBook, images_ds=images_ds)
-        app.app()
+# if __name__ == "__main__":
+
+# start the app
+st.set_page_config(page_title="Model Coffer Gallery", page_icon="🖼️", layout="wide")
+
+# remove ranking in the session state if it is created in Ranking.py
+st.session_state.pop('ranking', None)
+
+if 'user_id' not in st.session_state:
+    st.warning('Please log in first.')
+    home_btn = st.button('Go to Home Page')
+    if home_btn:
+        switch_page("home")
+else:
+    st.write('You have already logged in as ' + st.session_state.user_id[0])
+    roster, promptBook, images_ds = load_hf_dataset()
+    # print(promptBook.columns)
+
+    # initialize selected_dict
+    if 'selected_dict' not in st.session_state:
+        st.session_state['selected_dict'] = {}
+
+    app = GalleryApp(promptBook=promptBook, images_ds=images_ds)
+    app.app()