Update app.py

app.py

@@ -1,13 +1,16 @@
 import gradio as gr
 import pickle
 import numpy as np
-model  = pickle.load(open("model.pkl","rb"))
 
-def predict_v(input_value):
+def test(k):
     try:
-        if input_value == None: pass
-    except:
-        return np.argmax(model.predict(input_value / 255.))
-gr.Interface(fn=predict_v,
-            inputs = gr.SketchPad(),
-            outputs="number").launch()
+        if k == None:
+            pass
+    except: 
+        print(k.shape)
+        k = k / 255.
+        return np.argmax(model.predict(k))
+gr.Interface(inputs=gr.Sketchpad(),
+            outputs="number",
+            fn=test,
+            live=True).launch()