Update app.py

app.py

@@ -113,7 +113,7 @@ import numpy as np
 import gradio as gr
 
 def predict_sketch(img):
-  img_3d=img.reshape(-1,28,28)
+  img_3d=img.reshape(-1,28,28,1)
   im_resize=img_3d/255.0
   prediction=nn.predict(im_resize).tolist()[0]
   return {str(i):prediction[i] for i in range(10)}