Format

lm-sys · Nov 28, 2024 · 2ef314b · 2ef314b
1 parent d7a152a
commit 2ef314b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/fastchat/serve/gradio_block_arena_vision_anony.py b/fastchat/serve/gradio_block_arena_vision_anony.py
@@ -333,7 +333,9 @@ def add_text(
 
     # Use the first state to get the moderation response because this is based on user input so it is independent of the model
     moderation_image_input = images[0] if len(images) > 0 else None
-    moderation_type_to_response_map = states[0].content_moderator.image_and_text_moderation_filter(
+    moderation_type_to_response_map = states[
+        0
+    ].content_moderator.image_and_text_moderation_filter(
         moderation_image_input, text, model_list, do_moderation=True
     )
     text_flagged, nsfw_flag, csam_flag = (

diff --git a/fastchat/serve/gradio_block_arena_vision_named.py b/fastchat/serve/gradio_block_arena_vision_named.py
@@ -257,7 +257,9 @@ def add_text(
 
     # Use the first state to get the moderation response because this is based on user input so it is independent of the model
     moderation_image_input = images[0] if len(images) > 0 else None
-    moderation_type_to_response_map = states[0].content_moderator.image_and_text_moderation_filter(
+    moderation_type_to_response_map = states[
+        0
+    ].content_moderator.image_and_text_moderation_filter(
         moderation_image_input, text, model_list, do_moderation=False
     )
     text_flagged, nsfw_flag, csam_flag = (