Upload folder using huggingface_hub

app.py

@@ -62,7 +62,12 @@ def load_pipe():
         pipe = AutoPipelineForText2Image.from_pretrained(
             MODEL_ID,
             torch_dtype=_dtype,
+            safety_checker=None,
         )
+        if hasattr(pipe, "safety_checker"):
+            pipe.safety_checker = None
+        if hasattr(pipe, "requires_safety_checker"):
+            pipe.requires_safety_checker = False
         if hasattr(pipe, "enable_attention_slicing"):
             pipe.enable_attention_slicing()
         if hasattr(pipe, "vae") and hasattr(pipe.vae, "enable_tiling"):