ui adjustments

pages/Gallery.py

@@ -239,7 +239,7 @@ class GalleryApp:
 
             # items = items[items['prompt'] == selected_prompt].reset_index(drop=True)
 
-            st.title('Model Visualization and Retrieval')
+            # st.title('Model Visualization and Retrieval')
 
             # show source
             if isinstance(note, str):

pages/Ranking.py

@@ -26,8 +26,8 @@ class RankingApp:
 
     def sidebar(self, selected_prompt, items):
         with st.sidebar:
-            st.title('Personal Image Ranking')
-            st.write('Here you can test out your selected images with any prompt you like. ')
+            # st.title('Personal Image Ranking')
+            # st.write('Here you can test out your selected images with any prompt you like. ')
             # prompt_tags = self.promptBook['tag'].unique()
             # prompt_tags = np.sort(prompt_tags).tolist()
             #
@@ -264,7 +264,7 @@ class RankingApp:
         self.battle_images(items, prompt_id)
 
     def app(self):
-        # st.title('Personal Image Ranking')
+        st.write('### Personal Image Ranking')
         # st.write('Here you can test out your selected images with any prompt you like. ')
         # st.write(self.promptBook)
 
@@ -282,7 +282,7 @@ class RankingApp:
                                                                                   'tag'] in prompt_tags else 0
         print(tag_idx)
 
-        tag = st.radio('Select a tag', prompt_tags, index=tag_idx, horizontal=True)
+        tag = st.radio('Select a tag', prompt_tags, index=tag_idx, horizontal=True, label_visibility='collapsed')
         items = self.promptBook[self.promptBook['tag'] == tag].reset_index(drop=True)
         prompts = np.sort(items['prompt'].unique())[::-1].tolist()