Spaces:

simayhosmeyve
/

Image_Enhancement

Build error

simayhosmeyve commited on Mar 23, 2022

Commit

2216322

•

1 Parent(s): 027cbfd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -513,7 +513,7 @@ def result(Input,Choice):
       return Input
     if Choice=="Enhancement":
-      pre_trained2 = tf.keras.models.load_model("generatorLR-HR_300.h5")
       size0 = Input.shape[0]
       size1 = Input.shape[1]
       Input = cv2.resize(Input, (256,256), interpolation = cv2.INTER_AREA)

       return Input
     if Choice=="Enhancement":
+      pre_trained2 = tf.keras.models.load_model("gradio_pix2pix.h5")
       size0 = Input.shape[0]
       size1 = Input.shape[1]
       Input = cv2.resize(Input, (256,256), interpolation = cv2.INTER_AREA)