Update app.py

app.py

@@ -182,7 +182,7 @@ def result(Input,Choice,Step):
         size0 = Input.shape[0]
         size1 = Input.shape[1]
         Input = cv2.resize(Input, (256,256), interpolation = cv2.INTER_AREA)
-        Input = cv2.cvtColor(Input , cv2.COLOR_BGR2RGB)
+        Input = cv2.cvtColor(Input ,cv2.COLOR_BGR2RGB)
         Input = (Input/127.5) - 1
         Input = Input.astype(np.float32)
         Input = np.array(Input).reshape(1,256,256,3)
@@ -191,6 +191,7 @@ def result(Input,Choice,Step):
         Input = (Input+1)*127.5
         Input = np.uint8(Input)
         Input = cv2.resize(Input, (size1,size0), interpolation = cv2.INTER_AREA)
+        Input = cv2.cvtColor(Input ,cv2.COLOR_BGR2RGB)
         Psnr = 50
         return Input, Psnr