Update app.py

app.py

@@ -113,6 +113,7 @@ def decode_batch_predictions(input_image, img_size=(image_width, image_height)):
     
     input_image = tf.io.decode_image(input_image, channels=1,  dtype=tf.dtypes.uint8) 
     input_image = distortion_free_resize(input_image, img_size)
+    input_image = tf.expand_dims(input_image, axis=0)
     input_image = tf.image.convert_image_dtype(input_image, tf.float32)/255.0
     
     pred = loaded_model.predict(input_image)