Update app.py

app.py

@@ -9,7 +9,7 @@ model = load_model("./MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr}
 
 def predict_input_image(img):
   #img_4d=img.reshape(-1,180,180,3)
-  img_4d=img.reshape(256,256,3)
+  img_4d=img.reshape(-1,256,256,3)
   prediction=model.predict(img_4d)[0]
   return {flower_classes[i]: float(prediction[i]) for i in range(5)}