remove nsfw filter option in gallery mode

pages/Gallery.py

@@ -122,7 +122,8 @@ class GalleryApp:
         selecters = st.columns([1, 4])
 
         if 'score_weights' not in st.session_state:
-            st.session_state.score_weights = [1.0, 0.8, 0.2, 0.8]
+            # st.session_state.score_weights = [1.0, 0.8, 0.2, 0.8]
+            st.session_state.score_weights = [1.0, 0.8, 0.2]
 
         # select sort type
         with selecters[0]:
@@ -133,7 +134,7 @@ class GalleryApp:
         # select other options
         with selecters[1]:
             if sort_type == 'IDs and Names':
-                sub_selecters = st.columns([3, 1])
+                sub_selecters = st.columns([3])
                 # select sort by
                 with sub_selecters[0]:
                     sort_by = st.selectbox('Sort by',
@@ -144,7 +145,7 @@ class GalleryApp:
 
             else:
                 # add custom weights
-                sub_selecters = st.columns([1, 1, 1, 1])
+                sub_selecters = st.columns([1, 1, 1])
 
                 with sub_selecters[0]:
                     clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=1.0, step=0.1, help='the weight for normalized clip score')
@@ -163,13 +164,13 @@ class GalleryApp:
                 st.session_state.score_weights[1] = round(mcos_weight, 2)
                 st.session_state.score_weights[2] = round(pop_weight, 2)
 
-            # select threshold
-            with sub_selecters[continue_idx]:
-                nsfw_threshold = st.number_input('NSFW Score Threshold', min_value=0.0, max_value=1.0, value=0.8, step=0.01, help='Only show models with nsfw score lower than this threshold, set 1.0 to show all images')
-                items = items[items['norm_nsfw'] <= nsfw_threshold].reset_index(drop=True)
-
-            # save latest threshold
-            st.session_state.score_weights[3] = nsfw_threshold
+            # # select threshold
+            # with sub_selecters[continue_idx]:
+            #     nsfw_threshold = st.number_input('NSFW Score Threshold', min_value=0.0, max_value=1.0, value=0.8, step=0.01, help='Only show models with nsfw score lower than this threshold, set 1.0 to show all images')
+            #     items = items[items['norm_nsfw'] <= nsfw_threshold].reset_index(drop=True)
+            #
+            # # save latest threshold
+            # st.session_state.score_weights[3] = nsfw_threshold
 
         # # draw a distribution histogram
         # if sort_type == 'Scores':
@@ -495,10 +496,7 @@ class GalleryApp:
         #     refresh_btn = st.button('Refresh', on_click=gallery_space.empty, use_container_width=True)
 
         with gallery_space.container():
-            with st.spinner('Loading images...'):
-                self.gallery_standard(items, col_num, info)
-
-                # st.info("Don't forget to scroll back to top and click the 'Confirm Selection' button to save your selection!!!")
+            self.gallery_standard(items, col_num, info)
 
     def submit_actions(self, status, prompt_id):
         # remove counter from session state