Correct safety filter (#7)

- Update README.md (aebf2be7d6cdb3419c04f6d4631a4a9aa43708b5)


Co-authored-by: Apolinário from multimodal AI art <[email protected]>

README.md

@@ -36,6 +36,18 @@ Developers and creatives looking to build on top of `FLUX.1 Kontext [dev]` are e
 
 `FLUX.1 Kontext [dev]` is also available in both [ComfyUI](https://github.com/comfyanonymous/ComfyUI) and [Diffusers](https://github.com/huggingface/diffusers).
 
+## API Endpoints
+The FLUX.1 Kontext models are also available via API from the following sources
+- bfl.ai: https://docs.bfl.ai/
+- DataCrunch: https://datacrunch.io/flux-kontext
+- fal: https://fal.ai/flux-kontext
+- Replicate: https://replicate.com/blog/flux-kontext
+    - https://replicate.com/black-forest-labs/flux-kontext-dev
+    - https://replicate.com/black-forest-labs/flux-kontext-pro
+    - https://replicate.com/black-forest-labs/flux-kontext-max
+- Runware: https://runware.ai/blog/introducing-flux1-kontext-instruction-based-image-editing-with-ai?utm_source=bfl
+- TogetherAI: https://www.together.ai/models/flux-1-kontext-dev
+
 ### Using with diffusers 🧨
 
 ```shell
@@ -55,23 +67,9 @@ pipe.to("cuda")
 input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
 
 image = pipe(
-image=input_image,
-prompt="Add a hat to the cat",
-guidance_scale=2.5,
-).images[0]
-```
-
-Text-to-image:
-```py
-import torch
-from diffusers import FluxKontextPipeline
-
-pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16)
-pipe.to("cuda")
-
-image = pipe(
-prompt="A dog eating pizza",
-guidance_scale=2.5,
+  image=input_image,
+  prompt="Add a hat to the cat",
+  guidance_scale=2.5
 ).images[0]
 ```
 
@@ -80,9 +78,9 @@ Flux Kontext comes with an integrity checker, which should be run after the imag
 ```python
 import torch
 import numpy as np
-from flux.safety import PixtralIntegrity
+from flux.content_filters import PixtralContentFilter
 
-integrity_checker = PixtralIntegrity(torch.device("cuda"))
+integrity_checker = PixtralContentFilter(torch.device("cuda"))
 image_ = np.array(image) / 255.0
 image_ = 2 * image_ - 1
 image_ = torch.from_numpy(image_).to("cuda", dtype=torch.float32).unsqueeze(0).permute(0, 3, 1, 2)
@@ -92,17 +90,6 @@ raise ValueError("Your image has been flagged. Choose another prompt/image or tr
 
 For VRAM saving measures and speed ups check out the [diffusers docs](https://huggingface.co/docs/diffusers/en/index)
 
-## API Endpoints
-The FLUX.1 Kontext models are also available via API from the following sources
-- bfl.ai: https://docs.bfl.ai/
-- DataCrunch: https://datacrunch.io/flux-kontext
-- fal: https://fal.ai/flux-kontext
-- Replicate: https://replicate.com/blog/flux-kontext
-    - https://replicate.com/black-forest-labs/flux-kontext-dev
-    - https://replicate.com/black-forest-labs/flux-kontext-pro
-    - https://replicate.com/black-forest-labs/flux-kontext-max
-- Runware: https://runware.ai/blog/introducing-flux1-kontext-instruction-based-image-editing-with-ai?utm_source=bfl
-- TogetherAI: https://www.together.ai/models/flux-1-kontext-dev
 
 ---