niulx commited on
Commit
cda7518
1 Parent(s): 88dbfdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -173,7 +173,11 @@ import shutil
173
  if os.path.isdir("./example_tmp"):
174
  shutil.rmtree("./example_tmp")
175
 
 
176
  from segment import run_segmentation
 
 
 
177
  with gr.Blocks() as demo:
178
  image = gr.State() # store mask
179
  image_loaded = gr.State()
@@ -204,8 +208,9 @@ with gr.Blocks() as demo:
204
  # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
205
  mask_np_list_updated = mask_np_list
206
  with gr.Column():
 
207
  gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
208
- slider = gr.Slider(0, 20, step=1, label = 'mask id', interactive=False)
209
  label = gr.Textbox()
210
  slider.release(slider_release,
211
  inputs = [slider, image_loaded, mask_np_list_updated, mask_label_list],
@@ -214,7 +219,7 @@ with gr.Blocks() as demo:
214
 
215
  segment_button.click(run_segmentation,
216
  [canvas] ,
217
- [text_button] )
218
  text_button.click(load_image_ui, [false] ,
219
  [image_loaded, segmentation, mask_np_list, mask_label_list, canvas, slider] )
220
 
@@ -302,7 +307,8 @@ with gr.Blocks() as demo:
302
  gr.Markdown("""<p style="text-align: center; font-size: 20px">Editing setting (SD)</p>""")
303
 
304
  tgt_prompt = gr.Textbox(value="White bag", label="Editing: Text prompt", interactive= True )
305
- tgt_index = gr.Number(value="0", label="Editing: Object index", interactive= True )
 
306
  guidance_scale = gr.Textbox(value="6", label="Editing: CFG guidance scale", interactive= True )
307
  num_sampling_steps = gr.Number(value="50", label="Editing: Sampling steps", interactive= True )
308
  edge_thickness = gr.Number(value="10", label="Editing: Edge thickness", interactive= True )
@@ -342,7 +348,7 @@ with gr.Blocks() as demo:
342
  strength ,
343
  edge_thickness,
344
  tgt_prompt ,
345
- tgt_index
346
  ],
347
  outputs = [canvas_text_edit],queue=True,
348
  )
 
173
  if os.path.isdir("./example_tmp"):
174
  shutil.rmtree("./example_tmp")
175
 
176
+
177
  from segment import run_segmentation
178
+
179
+ slider = gr.Slider(0, 20, step=1, label = 'mask id', interactive=False)
180
+
181
  with gr.Blocks() as demo:
182
  image = gr.State() # store mask
183
  image_loaded = gr.State()
 
208
  # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
209
  mask_np_list_updated = mask_np_list
210
  with gr.Column():
211
+ result_info0 = gr.Text(label="Response")
212
  gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
213
+ slider.render()
214
  label = gr.Textbox()
215
  slider.release(slider_release,
216
  inputs = [slider, image_loaded, mask_np_list_updated, mask_label_list],
 
219
 
220
  segment_button.click(run_segmentation,
221
  [canvas] ,
222
+ [text_button,result_info0] )
223
  text_button.click(load_image_ui, [false] ,
224
  [image_loaded, segmentation, mask_np_list, mask_label_list, canvas, slider] )
225
 
 
307
  gr.Markdown("""<p style="text-align: center; font-size: 20px">Editing setting (SD)</p>""")
308
 
309
  tgt_prompt = gr.Textbox(value="White bag", label="Editing: Text prompt", interactive= True )
310
+ slider.render()
311
+ #tgt_index = gr.Number(value="0", label="Editing: Object index", interactive= True )
312
  guidance_scale = gr.Textbox(value="6", label="Editing: CFG guidance scale", interactive= True )
313
  num_sampling_steps = gr.Number(value="50", label="Editing: Sampling steps", interactive= True )
314
  edge_thickness = gr.Number(value="10", label="Editing: Edge thickness", interactive= True )
 
348
  strength ,
349
  edge_thickness,
350
  tgt_prompt ,
351
+ slider
352
  ],
353
  outputs = [canvas_text_edit],queue=True,
354
  )