Update app.py

app.py

@@ -7,12 +7,11 @@ def psnr(y_true, y_pred):
 from keras.models import load_model
 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(-1,256,256,3)
-  prediction=model.predict(img_4d)[0]
-  return {flower_classes[i]: float(prediction[i]) for i in range(5)}
+decoded_imgs = model.predict(image)
+
+#decoded_imgs.reshape(256,256,3)
+#prediction=model.predict(img_4d)[0]
 
 image = gr.inputs.Image(shape=(256,256))
 
-gr.Interface(fn=predict_input_image, inputs=image, outputs=image,interpretation='default').launch(debug='True')
+gr.Interface(fn=predict_input_image, inputs=image, outputs=decoded_imgs,interpretation='default').launch(debug='True')