multimodalart HF Staff commited on
Commit
5e398c6
·
verified ·
1 Parent(s): cd5feb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -194,6 +194,10 @@ def extract_info(json_data: Dict[str, Any], hunyuan_type: Optional[str] = None)
194
  if not image_url:
195
  continue
196
 
 
 
 
 
197
  filename_part = os.path.basename(image_url)
198
  image_id_str = filename_part.split('.')[0]
199
 
 
194
  if not image_url:
195
  continue
196
 
197
+ image_nsfw_level = image_obj.get("nsfwLevel", 0)
198
+ if image_nsfw_level > 5:
199
+ continue
200
+
201
  filename_part = os.path.basename(image_url)
202
  image_id_str = filename_part.split('.')[0]
203