haritsahm commited on
Commit
5d412e5
·
1 Parent(s): 7011045

Hotfix missing examples on docker and filter files

Browse files
Files changed (2) hide show
  1. .dockerignore +1 -0
  2. main.py +4 -1
.dockerignore CHANGED
@@ -2,5 +2,6 @@
2
  !models/
3
  !weights/
4
  !utils/
 
5
  !main.py
6
  !requirements.txt
 
2
  !models/
3
  !weights/
4
  !utils/
5
+ !examples/
6
  !main.py
7
  !requirements.txt
main.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  from pathlib import Path
3
  from typing import List
4
 
@@ -27,7 +28,9 @@ SUPPORTED_IMG_EXT = ['.png', '.jpg', '.jpeg']
27
 
28
  EXAMPLE_IMAGES = [
29
  os.path.join(os.path.dirname(__file__), f'examples/{f}')
30
- for f in os.listdir('examples/')]
 
 
31
 
32
 
33
  def filter_files(files: List) -> List:
 
1
  import os
2
+ import re
3
  from pathlib import Path
4
  from typing import List
5
 
 
28
 
29
  EXAMPLE_IMAGES = [
30
  os.path.join(os.path.dirname(__file__), f'examples/{f}')
31
+ for f in os.listdir('examples/')
32
+ if re.search(r'.*.(jpg|jpeg|tiff|JPEG|PNG|png)', f)
33
+ ]
34
 
35
 
36
  def filter_files(files: List) -> List: