added prediction

app.py

@@ -1,7 +1,17 @@
 import gradio as gr
+from fastai.vision.all import *
 
 def greet(name):
     return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+learn = load_learner('export.pkl')
+
+def pre(image):
+  pilim= PILImageBW.create(image)
+  t = learn.dls.test_dl([pilim], rm_type_tfms=None, num_workers=0)
+  p,_,d= learn.get_preds(dl=t,with_decoded=True)
+  return p.argmax().item()
+
+
+iface = gr.Interface(fn=pre, inputs="sketchpad", outputs="text")
 iface.launch()