Update app.py

app.py

@@ -73,12 +73,10 @@ 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_preprocessed
+    return image
 
 
 # def image_prediction(image, model):
@@ -108,11 +106,11 @@ 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())
-            # right_column.title("Prediction")
-            # right_column.write(predicted_idx)
+            outputs = model.predict(int(image_pred))
+            _, y_hat = outputs.max(1)
+            predicted_idx = str(y_hat.item())
+            right_column.title("Prediction")
+            right_column.write(predicted_idx)
             right_column.write(decode_predictions(outputs, top=2)[0])