update minor

gradio_artist_classifier.py

@@ -27,8 +27,10 @@ from gradcam_utils import get_img_4d_array, make_gradcam_heatmap, align_image_wi
 # ---------- Settings ----------
 ARTIST_META = 'artist.json'
 TREND_META = 'trend.json'
-EXAMPLES = ['monet.jpg']
+EXAMPLES = ['monet2.jpg', 'picasso.jpeg', 'lichtenstein_popart.jpg']
 ALPHA = 0.9
+IMG_WIDTH = 299
+IMG_HEIGHT = 299
 
 # ---------- Logging ----------
 logger.add('app.log', mode='a')
@@ -59,7 +61,7 @@ def resize_image(img_array, width, height):
 
 def predict(input_image):
     img_3d_array = load_image_as_array(input_image)
-    img_3d_array = resize_image(img_3d_array)
+    img_3d_array = resize_image(img_3d_array, IMG_WIDTH, IMG_HEIGHT)
     img_4d_array = img_3d_array[np.newaxis,...]
     logger.info(f'--- {input_image} loaded')