Update app.py

app.py

@@ -66,7 +66,7 @@ class_labels = ['battery',
 # Function to predict image label and score
 def predict_image(input):
     # Resize the image to the size expected by the model
-    image = input.resize((None, 224, 244, 3))
+    image = input.resize((224, 244))
     # Convert the image to a NumPy array
     image_array = tf.keras.preprocessing.image.img_to_array(image)
     # Normalize the image