adding a function to rescale the image

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import tensorflow as tf
 from load_model import build
 
+IMG_SHAPE = (224,224,3)
 
 
 def custom_standardization(s):
@@ -14,9 +15,11 @@ def custom_standardization(s):
     
 model = build()
 
+rescale = lambda image : tf.image.resize(tf.convert_to_tensor(image), IMG_SHAPE[:-1])
+
 def single_img_transcribe(image, temperature=1):
     initial = model.word_to_index([['[START]']]) # (batch, sequence)
-    img_features = model.feature_extractor(tf.convert_to_tensor(image)[tf.newaxis, ...])
+    img_features = model.feature_extractor(rescale(image)[tf.newaxis, ...])
 
     tokens = initial # (batch, sequence)
     for n in range(50):