Commit
•
61c5833
1
Parent(s):
7e062a4
Update app.py
Browse files
app.py
CHANGED
@@ -24,11 +24,11 @@ def predict(input_image):
|
|
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(
|
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()
|