ammariii08 commited on
Commit
62105f2
·
verified ·
1 Parent(s): 5fd20fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -492,7 +492,7 @@ def predict(
492
  # ---------------------
493
  t = time.time()
494
  reference_obj_img = make_square(reference_obj_img)
495
- reference_square_mask = remove_bg_u2netp(reference_obj_img)
496
  print("Reference image processing completed in {:.2f} seconds".format(time.time() - t))
497
 
498
  t = time.time()
 
492
  # ---------------------
493
  t = time.time()
494
  reference_obj_img = make_square(reference_obj_img)
495
+ reference_square_mask = remove_bg(reference_obj_img)
496
  print("Reference image processing completed in {:.2f} seconds".format(time.time() - t))
497
 
498
  t = time.time()