Update app.py

app.py

@@ -1,18 +1,21 @@
 import gradio as gr
-import pickle
 import numpy as np
 from modelutil import create_model
 
-def test(k):
+
+def predict_digit(image):
     try:
-        if k == None:
-            pass
-    except: 
-        print(k.shape)
-        k = k / 255.
+        if image == None: pass
+    except:
         model = create_model()
-        return np.argmax(model.predict(k.reshape(-1,28,28)))
-gr.Interface(inputs=gr.Sketchpad(),
-            outputs="number",
-            fn=test,
-            live=True).launch()
+        predictions = model.predict(image.reshape(1, 28, 28))
+        return np.argmax(predictions)
+
+gr.Interface(
+    title="MNIST Digit Classifier",
+    fn=predict_digit, 
+    inputs=gr.Sketchpad(input_shape=(28, 28), label="Draw a digit"),
+    outputs="number",
+    live=True
+).launch()
+