Update app.py

app.py

@@ -518,7 +518,7 @@ def result(Input,Choice):
       size1 = Input.shape[1]
       Input = cv2.resize(Input, (256,256), interpolation = cv2.INTER_AREA)
       Input = cv2.cvtColor(Input , cv2.COLOR_BGR2RGB)
-      Input = np.array(Input).reshape(1,256,256,3)
+      Input = np.array(Input).reshape(1,256,256,1)
       prediction = pre_trained2(Input,training=True)
       Input = prediction[0]
       Input = (Input+1)*127.5