Spaces:
Sleeping
Sleeping
turn off summary safety check
Browse files- pages/Summary.py +9 -9
pages/Summary.py
CHANGED
@@ -151,15 +151,15 @@ class DashboardApp:
|
|
151 |
# with st.expander(f'Show Images'):
|
152 |
images = self.promptBook[self.promptBook['modelVersion_id'] == modelVersion_id]['image_id'].values
|
153 |
|
154 |
-
safety_check = st.toggle('Include potentially unsafe or offensive images', value=False, key=modelVersion_id)
|
155 |
-
unsafe_prompts = json.load(open('data/unsafe_prompts.json', 'r'))
|
156 |
-
# merge dict values into one list
|
157 |
-
unsafe_prompts = [item for sublist in unsafe_prompts.values() for item in sublist]
|
158 |
-
unsafe_images = self.promptBook[self.promptBook['prompt_id'].isin(unsafe_prompts)]['image_id'].values
|
159 |
-
|
160 |
-
if not safety_check:
|
161 |
-
|
162 |
-
|
163 |
|
164 |
images = [f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{image}.png" for image in images]
|
165 |
clickable_images(
|
|
|
151 |
# with st.expander(f'Show Images'):
|
152 |
images = self.promptBook[self.promptBook['modelVersion_id'] == modelVersion_id]['image_id'].values
|
153 |
|
154 |
+
# safety_check = st.toggle('Include potentially unsafe or offensive images', value=False, key=modelVersion_id)
|
155 |
+
# unsafe_prompts = json.load(open('data/unsafe_prompts.json', 'r'))
|
156 |
+
# # merge dict values into one list
|
157 |
+
# unsafe_prompts = [item for sublist in unsafe_prompts.values() for item in sublist]
|
158 |
+
# unsafe_images = self.promptBook[self.promptBook['prompt_id'].isin(unsafe_prompts)]['image_id'].values
|
159 |
+
#
|
160 |
+
# if not safety_check:
|
161 |
+
# # exclude unsafe prompts from images
|
162 |
+
# images = [image for image in images if image not in unsafe_images]
|
163 |
|
164 |
images = [f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{image}.png" for image in images]
|
165 |
clickable_images(
|