Update app.py

app.py

@@ -179,6 +179,9 @@ def result(Input,Choice,Step):
     if Choice =="Enhancement":
       if Step == 1.0 or Step == 2.0 or Step == 3.0:
         pre_trained2 = tf.keras.models.load_model("generatorLR-HR_300.h5")
+        Input = cv2.cvtColor(Input , cv2.COLOR_BGR2RGB)
+        Input = (Input/127.5) - 1
+        Input = Input.astype(np.float32)
         size0 = Input.shape[0]
         size1 = Input.shape[1]
         Input = cv2.resize(Input, (256,256), interpolation = cv2.INTER_AREA)
@@ -186,7 +189,7 @@ def result(Input,Choice,Step):
         Input = np.array(Input).reshape(1,256,256,3)
         prediction = pre_trained2(Input,training=True)
         Input = prediction[0]
-        #Input = (Input+1)*127.5
+        Input = (Input+1)*127.5
         Input = np.uint8(Input)
         Input = cv2.resize(Input, (size1,size0), interpolation = cv2.INTER_AREA)
         Psnr = 50