hiding image ranking ui

Home.py

@@ -42,7 +42,7 @@ if __name__ == '__main__':
 
     st.set_page_config(page_title="Login", page_icon="🏠", layout="wide")
     st.write('A Research by MAPS Lab, NYU Shanghai')
-    st.title("Personalized Model Coffer")
+    st.title("Welcome to GEMRec Gallery Webapp!")
     st.write(
         "This is an web application to collect personal preference to ai generated images. \
         You can know which model you like most after you finish the survey."

pages/Ranking.py

@@ -134,19 +134,20 @@ if __name__ == "__main__":
             # st.write('You have checked ' + str(len(selected_modelVersions)) + ' images.')
             roster, promptBook, images_ds = load_hf_dataset()
             print(st.session_state.selected_dict)
-            # st.write("## roster")
-            # st.write(roster[roster['modelVersion_id'].isin(selected_modelVersions)])
-            # st.write(roster)
+            st.write("# Full function is coming soon.")
+            st.write("## roster")
+            st.write(roster[roster['modelVersion_id'].isin(selected_modelVersions)])
+            st.write(roster)
             # st.write("## promptBook")
             # st.write(promptBook)
 
-            # only select the part of the promptbook where tag is the same as st.session_state.selected_dict.keys(), while model version ids are the same as corresponding values to each key
-            promptBook_selected = pd.DataFrame()
-            for key, value in st.session_state.selected_dict.items():
-                promptBook_selected = promptBook_selected.append(promptBook[(promptBook['prompt_id'] == key) & (promptBook['modelVersion_id'].isin(value))])
-            promptBook_selected = promptBook_selected.reset_index(drop=True)
-            images_endpoint = "https://modelcofferbucket.s3-accelerate.amazonaws.com/"
-
-            app = RankingApp(promptBook_selected, images_endpoint, batch_size=4)
-            app.app()
+            # # only select the part of the promptbook where tag is the same as st.session_state.selected_dict.keys(), while model version ids are the same as corresponding values to each key
+            # promptBook_selected = pd.DataFrame()
+            # for key, value in st.session_state.selected_dict.items():
+            #     promptBook_selected = promptBook_selected.append(promptBook[(promptBook['prompt_id'] == key) & (promptBook['modelVersion_id'].isin(value))])
+            # promptBook_selected = promptBook_selected.reset_index(drop=True)
+            # images_endpoint = "https://modelcofferbucket.s3-accelerate.amazonaws.com/"
+            #
+            # app = RankingApp(promptBook_selected, images_endpoint, batch_size=4)
+            # app.app()