SkalskiP commited on
Commit
3e075bb
1 Parent(s): 03bb3c4

small mask fix

Browse files
Files changed (1) hide show
  1. app.py +9 -5
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(duration=200)
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=["#000000"], color_mode="fixed"))
58
- input_text_component = gr.Textbox(
59
- label='Text prompt',
60
- placeholder='Cartoon cactus',)
 
 
 
 
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():