Update app.py

app.py

@@ -177,18 +177,19 @@ def result(Input,Choice,Step):
       return Input,Psnr
       
     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)
-      Input = np.array(Input).reshape(1,256,256,3)
-      prediction = pre_trained2(Input,training=True)
-      Input = prediction[0]
-      Input = (Input+1)*127.5
-      Input = np.uint8(Input)
-      Input = cv2.resize(Input, (size1,size0), interpolation = cv2.INTER_AREA)
-      Psnr = 50
-      return Input, Psnr
+      if Step == 1.0 or Step == 2.0 or Step == 3.0:
+        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)
+        Input = np.array(Input).reshape(1,256,256,3)
+        prediction = pre_trained2(Input,training=True)
+        Input = prediction[0]
+        Input = (Input+1)*127.5
+        Input = np.uint8(Input)
+        #Input = cv2.resize(Input, (size1,size0))
+        Psnr = 50
+        return Input, Psnr