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

add new multiple file reader

Browse files
Files changed (1) hide show
  1. main.py +5 -2
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/')