Update app.py

app.py

@@ -182,12 +182,13 @@ 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_BGR2GRAY)
         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))
+        Input = cv2.resize(Input, (size1,size0), interpolation = cv2.INTER_AREA)
         Psnr = 50
         return Input, Psnr