justin-zk commited on
Commit
ab98f4d
1 Parent(s): ef7fe6c

update code

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -505,12 +505,12 @@ main_scribble = gr.Interface(
505
  fn=inference_scribble,
506
  inputs=[
507
  gr.ImageMask(label="[Stroke] Draw on Image", brush_radius=4),
508
- gr.Image(type="pil", label="test image1"),
509
- gr.Image(type="pil", label="test image2"),
510
  ],
511
  outputs=[
512
- gr.Image(type="pil", label="output image1").style(height=256, width=256),
513
- gr.Image(type="pil", label="output image2").style(height=256, width=256),
514
  ],
515
  allow_flagging="never",
516
  cache_examples=False,
 
505
  fn=inference_scribble,
506
  inputs=[
507
  gr.ImageMask(label="[Stroke] Draw on Image", brush_radius=4),
508
+ gr.Image(label="test image1"),
509
+ gr.Image(label="test image2"),
510
  ],
511
  outputs=[
512
+ gr.Image(label="output image1").style(height=256, width=256),
513
+ gr.Image(label="output image2").style(height=256, width=256),
514
  ],
515
  allow_flagging="never",
516
  cache_examples=False,