initial commit

.gitignore

@@ -0,0 +1,2 @@
+/venv
+/.idea

README.md

@@ -1,13 +1,13 @@
 ---
-title: FLUX.1 Inpaint
-emoji: 🏢
+title: FLUX.1 [Inpainting]
+emoji: 🎨
 colorFrom: yellow
-colorTo: yellow
+colorTo: pink
 sdk: gradio
-sdk_version: 4.41.0
+sdk_version: 4.40.0
 app_file: app.py
 pinned: false
-license: apache-2.0
+license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,77 @@
+import torch
+import gradio as gr
+from diffusers import FluxInpaintPipeline
+
+MARKDOWN = """
+# FLUX.1 Inpainting 🔥
+"""
+
+DEVICE = torch.device("cuda")
+# DEVICE = torch.device("cpu")
+
+pipe = FluxInpaintPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-schnell",
+    torch_dtype=torch.bfloat16)
+pipe.to(DEVICE)
+
+
+@spaces.GPU()
+def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True)):
+    if not input_text:
+        gr.Info("Please enter a text prompt.")
+        return None
+
+    image = input_image_editor['background']
+    mask_image = input_image_editor['layers'][0]
+
+    if not image:
+        gr.Info("Please upload an image.")
+        return None
+
+    if not mask_image:
+        gr.Info("Please draw a mask on the image.")
+        return None
+
+    generator = torch.Generator().manual_seed(42)
+    return pipe(
+        prompt=input_text,
+        image=image,
+        mask_image=mask_image,
+        width=1024,
+        height=1024,
+        strength=0.9
+    ).images[0]
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(MARKDOWN)
+    with gr.Row():
+        with gr.Column():
+            input_image_editor_component = gr.ImageEditor(
+                label='Image',
+                type='pil',
+                sources=["upload", "webcam"],
+                image_mode='RGB',
+                layers=False,
+                brush=gr.Brush(colors=["#000000"], color_mode="fixed"))
+            input_text_component = gr.Textbox(
+                label='Text prompt',
+                placeholder='Cartoon cactus',)
+            submit_button_component = gr.Button(
+                value='Submit', variant='primary')
+        with gr.Column():
+            output_image_component = gr.Image(
+                type='pil', image_mode='RGB', label='Generated image')
+
+    submit_button_component.click(
+        fn=process,
+        inputs=[
+            input_image_editor_component,
+            input_text_component
+        ],
+        outputs=[
+            output_image_component
+        ]
+    )
+
+demo.launch(debug=False, show_error=True)

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+spaces
+git+https://github.com/Gothos/diffusers.git@flux-inpaint