Update app.py

app.py

@@ -4,6 +4,8 @@ import numpy as np
 import random
 from keras.models import load_model
 import cv2
+from PIL import Image
+
 
 
 def psnr(y_true, y_pred):
@@ -22,7 +24,8 @@ def sepia(inp):
     sepia_img = sepia_img / 255.0
     sepia_img = model.predict(sepia_img)
     sepia_img = sepia_img*255
-    sepia_img = sepia_img.reshape(img_height, img_width, 3)
+    #sepia_img = sepia_img.reshape(img_height, img_width, 3)
+    sepia_img = Image.fromarray(sepia_img)
     return sepia_img
 
 demo = gr.Interface(fn=sepia, inputs=gr.inputs.Image(256,256),outputs="image").launch()