Update app.py

app.py

@@ -26,7 +26,7 @@ if day_inp is not None:
   img = Image.open(day_inp)
   img = img.resize((256,256))
   img = np.asarray(img)
-  img = np.reshape(img,(1,256,256,3))
+  img = np.reshape(img,(256,256,3))
   
   pred = model_out('FaceWithMask.h5', img)
   st.subheader('Input Image')