Update app.py

app.py

@@ -7,11 +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})
 
+image = gr.inputs.Image(shape=(256,256))
 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=decoded_imgs,interpretation='default').launch(debug='True')
+gr.Interface(inputs=image, outputs=decoded_imgs,interpretation='default').launch(debug='True')