Spaces:
Runtime error
Runtime error
small mask fix
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ pipe = FluxInpaintPipeline.from_pretrained(
|
|
| 17 |
"black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
| 18 |
|
| 19 |
|
| 20 |
-
@spaces.GPU(
|
| 21 |
def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True)):
|
| 22 |
if not input_text:
|
| 23 |
gr.Info("Please enter a text prompt.")
|
|
@@ -54,10 +54,14 @@ with gr.Blocks() as demo:
|
|
| 54 |
sources=["upload", "webcam"],
|
| 55 |
image_mode='RGB',
|
| 56 |
layers=False,
|
| 57 |
-
brush=gr.Brush(colors=["#
|
| 58 |
-
input_text_component = gr.
|
| 59 |
-
label=
|
| 60 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
submit_button_component = gr.Button(
|
| 62 |
value='Submit', variant='primary')
|
| 63 |
with gr.Column():
|
|
|
|
| 17 |
"black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
| 18 |
|
| 19 |
|
| 20 |
+
@spaces.GPU()
|
| 21 |
def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True)):
|
| 22 |
if not input_text:
|
| 23 |
gr.Info("Please enter a text prompt.")
|
|
|
|
| 54 |
sources=["upload", "webcam"],
|
| 55 |
image_mode='RGB',
|
| 56 |
layers=False,
|
| 57 |
+
brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
|
| 58 |
+
input_text_component = gr.Text(
|
| 59 |
+
label="Prompt",
|
| 60 |
+
show_label=False,
|
| 61 |
+
max_lines=1,
|
| 62 |
+
placeholder="Enter your prompt",
|
| 63 |
+
container=False,
|
| 64 |
+
)
|
| 65 |
submit_button_component = gr.Button(
|
| 66 |
value='Submit', variant='primary')
|
| 67 |
with gr.Column():
|