s194649 commited on
Commit
7f19a53
·
1 Parent(s): 6eb6376
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -137,12 +137,12 @@ with block:
137
 
138
  # event - init coords
139
  def on_reset_btn_click(input_image):
140
- return input_image, point_coords_empty(), point_labels_empty(), None, []
141
 
142
  reset_btn.click(
143
  on_reset_btn_click,
144
  [input_image],
145
- [input_image, point_coords, point_labels],
146
  queue=False,
147
  )
148
 
@@ -274,7 +274,7 @@ with block:
274
 
275
  def on_depth_reconstruction_mask_btn_click(inputs):
276
  print("depth reconstruction")
277
- path = dpt.generate_obj_masks(
278
  image=inputs[input_image],
279
  cube_size=inputs[cube_size],
280
  n_samples=inputs[n_samples],
 
137
 
138
  # event - init coords
139
  def on_reset_btn_click(input_image):
140
+ return ImageOps.mirror(input_image), point_coords_empty(), point_labels_empty(), None, []
141
 
142
  reset_btn.click(
143
  on_reset_btn_click,
144
  [input_image],
145
+ [prompt_image, point_coords, point_labels],
146
  queue=False,
147
  )
148
 
 
274
 
275
  def on_depth_reconstruction_mask_btn_click(inputs):
276
  print("depth reconstruction")
277
+ path = dpt.generate_obj_masks2(
278
  image=inputs[input_image],
279
  cube_size=inputs[cube_size],
280
  n_samples=inputs[n_samples],