Add input image resizer

main.py

@@ -18,6 +18,7 @@ BILATERAL_MODEL.load_state_dict(torch.load(
     BILATERIAL_WEIGHT, map_location='cpu'))
 BILATERAL_MODEL = BILATERAL_MODEL.to('cpu')
 BILATERAL_MODEL.eval()
+INPUT_HEIGHT, INPUT_WIDTH = 600, 500
 
 OUTPUT_GALLERY = gr.Gallery(
     label='Highlighted Area').style(grid=[2], height='auto')
@@ -127,6 +128,8 @@ def predict_bilateral(files):
         image = np.array(Image.open(str(path)))
         image = cv2.normalize(
             image, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)
+        image = cv2.resize(
+            image, (INPUT_WIDTH, INPUT_HEIGHT), interpolation=cv2.INTER_LINEAR)
 
         images.append(image)