yupikopi commited on
Commit
4ac0de6
·
1 Parent(s): e53b0ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ def psnr(y_true, y_pred):
3
  return -10*K.log(K.mean(K.flatten((y_true - y_pred))**2)) / np.log(10)
4
 
5
  from keras.models import load_model
6
- model = load_model("C:/Users/owner/Desktop/MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr})
7
 
8
  def predict_input_image(img):
9
  #img_4d=img.reshape(-1,180,180,3)
 
3
  return -10*K.log(K.mean(K.flatten((y_true - y_pred))**2)) / np.log(10)
4
 
5
  from keras.models import load_model
6
+ model = load_model("./MyNet.h5", custom_objects={'psnr': psnr, 'val_psnr': psnr})
7
 
8
  def predict_input_image(img):
9
  #img_4d=img.reshape(-1,180,180,3)