Spaces:
Runtime error
Runtime error
Commit
·
9d9900d
1
Parent(s):
96bab78
filter files
Browse files
app.py
CHANGED
@@ -34,7 +34,10 @@ def make_label_folders():
|
|
34 |
|
35 |
|
36 |
def predictions_into_folders(files):
|
37 |
-
files = [file.name for file in files
|
|
|
|
|
|
|
38 |
folders = make_label_folders()
|
39 |
predictions = pipe(files)
|
40 |
for file, prediction in zip(files, predictions):
|
@@ -48,9 +51,8 @@ def predictions_into_folders(files):
|
|
48 |
|
49 |
demo = gr.Interface(
|
50 |
predictions_into_folders,
|
51 |
-
gr.
|
52 |
gr.Files(),
|
53 |
cache_examples=True,
|
54 |
)
|
55 |
-
|
56 |
-
demo.launch(enable_queue=True)
|
|
|
34 |
|
35 |
|
36 |
def predictions_into_folders(files):
|
37 |
+
files = [file.name for file in files]
|
38 |
+
files = [
|
39 |
+
file for file in files if not file.startswith(".") and "DS_Store" not in file
|
40 |
+
]
|
41 |
folders = make_label_folders()
|
42 |
predictions = pipe(files)
|
43 |
for file, prediction in zip(files, predictions):
|
|
|
51 |
|
52 |
demo = gr.Interface(
|
53 |
predictions_into_folders,
|
54 |
+
gr.Files(file_count="directory", file_types=["image"]),
|
55 |
gr.Files(),
|
56 |
cache_examples=True,
|
57 |
)
|
58 |
+
demo.launch(debug=True)
|
|