Robledo Gularte Gonçalves commited on
Commit
95745c8
·
1 Parent(s): 8837d6b

add run segmentation logs

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -235,7 +235,10 @@ def run_full(image: str, req: gr.Request):
235
  @spaces.GPU()
236
  @torch.no_grad()
237
  def run_segmentation(image: str):
 
238
  image = prepare_image(image, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
 
 
239
  return image
240
 
241
  @spaces.GPU(duration=90)
 
235
  @spaces.GPU()
236
  @torch.no_grad()
237
  def run_segmentation(image: str):
238
+ print("run_segmentation pre!")
239
  image = prepare_image(image, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
240
+ print("run_segmentation pos!")
241
+ print("run_segmentation image: ", image)
242
  return image
243
 
244
  @spaces.GPU(duration=90)