Update app.py

app.py

@@ -7,7 +7,7 @@ def predict_v(input_value):
     try:
         if input_value == None: pass
     except:
-        return np.argmax(input_value / 255.)
+        return np.argmax(model.predict(input_value / 255.))
 gr.Interface(fn=predict_v,
             inputs = gr.SketchPad(),
             outputs="number").launch()