Update app.py

app.py

@@ -20,9 +20,13 @@ model = keras.models.load_model('best_model.h5')
 # Define a function that takes an image as input and uses the model for inference
 def image_classifier(image):
     # Preprocess the input image
+    image = Image.fromarray(image)
+    image = image.convert('L')
+    image = image.resize((128, 128))
     image = np.array(image)
     image = image / 255.0
     image = np.expand_dims(image, axis=0)
+    image = np.expand_dims(image, axis=-1)
     
     # Use your custom model for inference
     predictions = model.predict(image)