ahmedabdelwahed commited on
Commit
61c5833
1 Parent(s): 7e062a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,11 +24,11 @@ def predict(input_image):
24
  return mask
25
 
26
  def load_examples():
27
- files = os.listdir('examples/')
28
  img_list = []
29
  for file in files:
30
  if '.jpg' in file:
31
- img_list.append(str('examples/' + file))
32
  return img_list
33
 
34
  examples = load_examples()
 
24
  return mask
25
 
26
  def load_examples():
27
+ files = os.listdir()
28
  img_list = []
29
  for file in files:
30
  if '.jpg' in file:
31
+ img_list.append(file)
32
  return img_list
33
 
34
  examples = load_examples()