faster

backend/safety_check.py

@@ -13,4 +13,5 @@ def is_safe_image(
             nsfw_score = label["score"]
         elif label["label"] == "normal":
             normal_score = label["score"]
+    print(f"nsfw_score: {nsfw_score}, normal_score: {normal_score}")
     return normal_score > nsfw_score

frontend/webui/hf_demo.py

@@ -60,8 +60,8 @@ def predict(
     lcm_diffusion_setting.use_tiny_auto_encoder = False
     # lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
     # lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
-    lcm_diffusion_setting.image_width = 512
-    lcm_diffusion_setting.image_height = 512
+    lcm_diffusion_setting.image_width = 320
+    lcm_diffusion_setting.image_height = 320
     lcm_diffusion_setting.use_openvino = True
     lcm_diffusion_setting.use_tiny_auto_encoder = False
     pprint(lcm_diffusion_setting.model_dump())
@@ -75,7 +75,7 @@ def predict(
         classifier,
         result,
     ):
-        return result  # .resize([512, 512], PIL.Image.ANTIALIAS)
+        return result.resize([512, 512], Image.LANCZOS)
     else:
         print("Unsafe image detected")
         return Image.new("RGB", (512, 512), (0, 0, 0))