Upload app.py

app.py

@@ -57,6 +57,10 @@ pipe_i2i = AutoPipelineForImage2Image.from_pretrained(base_model,
                                                       torch_dtype=dtype
                                                      )
 
+# Disable the safety (censor) mechanism
+pipe.safety_checker = lambda images, clip_input, **kwargs: (images, False)
+pipe_i2i.safety_checker = lambda images, clip_input, **kwargs: (images, False)
+
 MAX_SEED = 2**32-1
 
 pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
@@ -289,7 +293,7 @@ css = '''
 font=[gr.themes.GoogleFont("Source Sans Pro"), "Arial", "sans-serif"]
 with gr.Blocks(theme=gr.themes.Soft(font=font), css=css, delete_cache=(60, 60)) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA the Explorer</h1>""",
+        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA Kayte's Space</h1>""",
         elem_id="title",
     )
     selected_index = gr.State(None)