Update main func to read from multiple file input

main.py

@@ -116,7 +116,7 @@ def filter_files(files: List) -> List:
     return file_paths
 
 
-def predict_bilateral(files):
+def predict_bilateral(cc_file, mlo_file):
     """Predict Bilateral Mammography.
 
     Parameters
@@ -130,7 +130,7 @@ def predict_bilateral(files):
         List of objects that will be used to display the result
     """
 
-    filtered_files = filter_files(files)
+    filtered_files = filter_files([cc_file, mlo_file])
 
     displays_imgs = []
     images = []
@@ -188,7 +188,7 @@ def run():
     """Run Gradio App."""
     demo = gr.Interface(
         fn=predict_bilateral,
-        inputs=gr.File(file_count='multiple', file_types=SUPPORTED_IMG_EXT),
+        inputs=INPUT_FILES,
         outputs=[OUTPUT_GALLERY, gr.Label(label='Cancer Type')],
         examples=EXAMPLE_IMAGES,
     )