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/')
|