Update app.py

app.py

@@ -9,13 +9,23 @@ from keras.models import load_model
 model = load_model("./MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr})
 
 # ラップする関数
-def sepia(input_img):
-    sepia_img = np.asarray(input_img)
-    sepia_img = sepia_img.astype('float32')
-    sepia_img = sepia_img / 255.0
-    sepia_img = model.predict(sepia_img)
+#def sepia(input_img):
+#    sepia_img = np.asarray(input_img)
+#    sepia_img = sepia_img.astype('float32')
+#    sepia_img = sepia_img / 255.0
+#    sepia_img = model.predict(sepia_img)
+#    return sepia_img
+
+def sepia(inp):
+    inp = inp.reshape((-1, 256, 256, 3))
+    inp = np.asarray(inp)
+    inp = inp.astype('float32')
+    inp = inp / 255.0
+    sepia_img = model.predict(inp)
     return sepia_img
 
+
+
 # シンプルなUIを作成
 demo = gr.Interface(
     fn=sepia,