haritsahm
commited on
Commit
·
3ae607f
1
Parent(s):
5d412e5
add new multiple file reader
Browse files
main.py
CHANGED
@@ -22,10 +22,13 @@ BILATERAL_MODEL = BILATERAL_MODEL.to('cpu')
|
|
22 |
BILATERAL_MODEL.eval()
|
23 |
INPUT_HEIGHT, INPUT_WIDTH = 600, 500
|
24 |
|
|
|
|
|
|
|
|
|
|
|
25 |
OUTPUT_GALLERY = gr.Gallery(
|
26 |
label='Highlighted Area').style(grid=[2], height='auto')
|
27 |
-
SUPPORTED_IMG_EXT = ['.png', '.jpg', '.jpeg']
|
28 |
-
|
29 |
EXAMPLE_IMAGES = [
|
30 |
os.path.join(os.path.dirname(__file__), f'examples/{f}')
|
31 |
for f in os.listdir('examples/')
|
|
|
22 |
BILATERAL_MODEL.eval()
|
23 |
INPUT_HEIGHT, INPUT_WIDTH = 600, 500
|
24 |
|
25 |
+
SUPPORTED_IMG_EXT = ['.png', '.jpg', '.jpeg']
|
26 |
+
INPUT_FILES = [
|
27 |
+
gr.File(file_count='single', file_types=SUPPORTED_IMG_EXT, label='CC View'),
|
28 |
+
gr.File(file_count='single', file_types=SUPPORTED_IMG_EXT, label='MLO View'),
|
29 |
+
]
|
30 |
OUTPUT_GALLERY = gr.Gallery(
|
31 |
label='Highlighted Area').style(grid=[2], height='auto')
|
|
|
|
|
32 |
EXAMPLE_IMAGES = [
|
33 |
os.path.join(os.path.dirname(__file__), f'examples/{f}')
|
34 |
for f in os.listdir('examples/')
|