Update app.py

app.py

@@ -9,14 +9,30 @@ 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))
-image = np.asarray(image)
+# ラップする関数
+def sepia(input_img):
+    sepia_img = np.asarray(sepia_img)
+    #image = image.astype('float32')
+    sepia_img = sepia_img / 255.0
+    sepia_img = model.predict(sepia_img)
+    return sepia_img
+
+# シンプルなUIを作成
+demo = gr.Interface(
+    fn=sepia, 
+    inputs=gr.Image(shape=(256, 256)), 
+    outputs="image"
+)
+
+
+#image = gr.inputs.Image(shape=(256,256))
+#image = np.asarray(image)
 #image = image.astype('float32')
-image = image / 255.0
-decoded_imgs = model.predict(image)
+#image = image / 255.0
+#decoded_imgs = model.predict(image)
 
 #decoded_imgs.reshape(256,256,3)
 #prediction=model.predict(img_4d)[0]
 
 
-gr.Interface(inputs=image, outputs=decoded_imgs,interpretation='default').launch(debug='True')
+#gr.Interface(inputs=image, outputs=decoded_imgs,interpretation='default').launch(debug='True')