Update app.py

app.py

@@ -12,7 +12,7 @@ f = st.file_uploader("Choose a Photo")
 if f is not None:
     img = Image.open(f)
     img = img.resize((256,256))
-    img = np.reshape(img,(1,28,28,3))
+    img = np.reshape(img,(1,256,256,3))
     pred,pred2 = model.predict(img),model2.predict(img)
     
     st.image(img,use_column_width=True)