xinlongwang commited on
Commit
bded866
·
1 Parent(s): 1602533
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -56,11 +56,11 @@ examples = [
56
  ['./images/ydt_2.jpg', './images/ydt_1.jpg', './images/ydt_3.jpg'],
57
  ]
58
 
59
- css = ".output-image, .input-image, .image-preview {height: 600px !important}"
60
  demo_mask = gr.Interface(fn=inference_mask1,
61
  inputs=[gr.ImageMask(label="prompt (提示图)"), gr.Image(label="img1 (测试图1)"), gr.Image(label="img2 (测试图2)")],
62
  #outputs=[gr.Image(shape=(448, 448), label="output1 (输出图1)"), gr.Image(shape=(448, 448), label="output2 (输出图2)")],
63
- outputs=[gr.Image(label="output1 (输出图1)"), gr.Image(label="output2 (输出图2)")],
64
  #outputs=gr.Gallery(label="outputs (输出图)"),
65
  examples=examples,
66
  #title="SegGPT for Any Segmentation<br>(Painter Inside)",
@@ -72,7 +72,7 @@ demo_mask = gr.Interface(fn=inference_mask1,
72
  </p>",
73
  cache_examples=False,
74
  allow_flagging="never",
75
- css=css,
76
  )
77
 
78
 
 
56
  ['./images/ydt_2.jpg', './images/ydt_1.jpg', './images/ydt_3.jpg'],
57
  ]
58
 
59
+ css = ".output-image, .input-image, .image-preview {height: 300px !important}"
60
  demo_mask = gr.Interface(fn=inference_mask1,
61
  inputs=[gr.ImageMask(label="prompt (提示图)"), gr.Image(label="img1 (测试图1)"), gr.Image(label="img2 (测试图2)")],
62
  #outputs=[gr.Image(shape=(448, 448), label="output1 (输出图1)"), gr.Image(shape=(448, 448), label="output2 (输出图2)")],
63
+ outputs=[gr.Image(label="output1 (输出图1)").style(height=400, width=400), gr.Image(label="output2 (输出图2)").style(height=400, width=400)],
64
  #outputs=gr.Gallery(label="outputs (输出图)"),
65
  examples=examples,
66
  #title="SegGPT for Any Segmentation<br>(Painter Inside)",
 
72
  </p>",
73
  cache_examples=False,
74
  allow_flagging="never",
75
+ css=css,
76
  )
77
 
78