Update app.py

app.py

@@ -72,10 +72,12 @@ def image_transformation(image):
     #image = Image._resize_dispatcher(image, new_shape=(256, 256))
     #image= np.resize((256,256))
     image = np.array(image)
+    img_batch = np.expand_dims(image, axis=0)
+    img_preprocessed = preprocess_input(img_batch)
     np.save('images.npy', image)
     image = np.load('images.npy', allow_pickle=True)
 
-    return image
+    return image_preprocessed
 
 
 # def image_prediction(image, model):
@@ -105,7 +107,6 @@ def main():
         # labels = ohe.fit_transform(label).toarray()
 
         if pred_button:
-            
             outputs = model.predict(image_pred)
             #_, y_hat = outputs.max(1)
             # predicted_idx = str(y_hat.item())