hylee commited on
Commit
420ecfb
1 Parent(s): ec88f2e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ def process(im):
93
  image = modnet.segment(im.name)
94
  Image.fromarray(np.uint8(image)).save(im.name)
95
 
96
- img_name_list = glob.glob(im.name)
97
  print("Number of images: ", len(img_name_list))
98
  # --------- 2. dataloader ---------
99
  # 1. dataloader
 
93
  image = modnet.segment(im.name)
94
  Image.fromarray(np.uint8(image)).save(im.name)
95
 
96
+ img_name_list = [im.name]
97
  print("Number of images: ", len(img_name_list))
98
  # --------- 2. dataloader ---------
99
  # 1. dataloader