add resize option

gradio_artist_classifier.py

@@ -13,7 +13,9 @@ import seaborn as sns
 import io
 import json
 import numpy as np
+import skimage
 import skimage.io
+from skimage.transform import resize
 from loguru import logger
 from huggingface_hub import from_pretrained_keras
 import gradio as gr
@@ -49,14 +51,15 @@ def load_image_as_array(image_file):
         img = img[..., :-1]
     return img
 
-def load_image_as_tensor(image_file):
-    img = tf.io.read_file(image_file)
-    img = tf.io.decode_jpeg(img, channels=3)
-    return img
+def resize_image(img_array, width, height):
+    img_resized = resize(img_array, (height, width), 
+                         anti_aliasing=True, 
+                         preserve_range=False)
+    return skimage.img_as_ubyte(img_resized)
 
 def predict(input_image):
     img_3d_array = load_image_as_array(input_image)
-    # img_4d_tensor = load_image_as_tensor(input_image)
+    img_3d_array = resize_image(img_3d_array)
     img_4d_array = img_3d_array[np.newaxis,...]
     logger.info(f'--- {input_image} loaded')