davanstrien HF Staff commited on
Commit
9d9900d
·
1 Parent(s): 96bab78

filter files

Browse files
Files changed (1) hide show
  1. app.py +6 -4
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 if not file.name.startswith(".")]
 
 
 
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.File(file_count="directory", file_types=["image"]),
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)