Spaces:

simayhosmeyve
/

Image_Enhancement

Build error

simayhosmeyve commited on Apr 3, 2022

Commit

49bd8dc

•

1 Parent(s): f78336f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -518,7 +518,7 @@ def result(Input,Choice,Step):
       if Step == 2:
         pre_trained = tf.keras.models.load_model("indoor_2.h5")
       if Step == 3:
-        pre_trained = tf.keras.models.load_model("indoor_1.h5")
       size0 = Input.shape[0]
       size1 = Input.shape[1]

       if Step == 2:
         pre_trained = tf.keras.models.load_model("indoor_2.h5")
       if Step == 3:
+        pre_trained = tf.keras.models.load_model("indoor_3.h5")
       size0 = Input.shape[0]
       size1 = Input.shape[1]