Abijith commited on
Commit
15bdaaf
1 Parent(s): f5b5ce5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -165,11 +165,15 @@ with gr.Blocks() as demo:
165
  """)
166
  with gr.Row():
167
  image_input = gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil')
168
- with gr.Row():
169
  image_output1 = gr.Image(type="pil", label="Mask with Image")
 
 
 
 
 
170
  image_output2 = gr.Image(type="pil", label="Mask")
171
-
172
- image_button = gr.Button("Genarate-Segment-Mask")
173
 
174
  image_button.click(inference_scribble, inputs=image_input, outputs=[image_output1, image_output2])
175
  image_input.upload(reset_data)
 
165
  """)
166
  with gr.Row():
167
  image_input = gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil')
 
168
  image_output1 = gr.Image(type="pil", label="Mask with Image")
169
+ with gr.Row():
170
+ examples = gr.Examples(examples=["./cardamage_example/0006.JPEG",
171
+ "./cardamage_example/0008.JPEG",
172
+ "./cardamage_example/0206.JPEG"],
173
+ inputs=image_input)
174
  image_output2 = gr.Image(type="pil", label="Mask")
175
+
176
+ image_button = gr.Button("Genarate-Segment-Mask", variant='primary')
177
 
178
  image_button.click(inference_scribble, inputs=image_input, outputs=[image_output1, image_output2])
179
  image_input.upload(reset_data)