yupikopi commited on
Commit
cc5377e
·
1 Parent(s): 93fa302

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,8 +18,8 @@ img_height = 256 # 画像の縦画素数
18
  model = load_model("./MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr})
19
 
20
  def sepia(inp):
21
- sepia_img = cv2.imread(inp)
22
- sepia_img = np.asarray(sepia_img)
23
  sepia_img = sepia_img.astype('float32')
24
  sepia_img = sepia_img / 255.0
25
  sepia_img = model.predict(sepia_img)
@@ -28,4 +28,4 @@ def sepia(inp):
28
  sepia_img = sepia_img.reshape(img_height, img_width, 3)
29
  return sepia_img
30
 
31
- demo = gr.Interface(fn=sepia, inputs=gr.inputs.Image(256,256),outputs=sepia_img).launch()
 
18
  model = load_model("./MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr})
19
 
20
  def sepia(inp):
21
+ #sepia_img = cv2.imread(inp)
22
+ sepia_img = np.asarray(inp)
23
  sepia_img = sepia_img.astype('float32')
24
  sepia_img = sepia_img / 255.0
25
  sepia_img = model.predict(sepia_img)
 
28
  sepia_img = sepia_img.reshape(img_height, img_width, 3)
29
  return sepia_img
30
 
31
+ demo = gr.Interface(fn=sepia, inputs=gr.inputs.Image(256,256),outputs="image").launch()