Upload 2 files

app.py

@@ -7,12 +7,14 @@ import matplotlib.pyplot as plt
 import numpy as np
 
 cust = {'InstanceNormalization': InstanceNormalization}
-model=load_model('g_model_AtoB_002160.h5',cust)
+# model=load_model('g_model_AtoB_002160.h5',cust)
+model=load_model('g_model_AtoB_000540.h5',cust)
+
 path  = [['ex1.jpg'], ['ex2.jpg']]
 
 def show_preds_image(image_path):
     A = plt.imread(image_path)
-    A = (A - 127.5) / 127.5
+    # A = (A - 127.5) / 127.5
     A = np.expand_dims(A,axis=0)
     B = model.predict(A)
     B = np.squeeze(B,axis=0)

g_model_AtoB_000540.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ad0952879f30273739e417543b75bcc5d04aa401d76e5a232dac7e8b78f31f3
+size 141314344