Update app.py

app.py

@@ -1,2 +1,16 @@
 from fastai.vision.all import *
 import gradio as gr
+
+learner = load_leaner("model.pkl")
+
+with open('categories.txt', 'r') as file:
+    categories = file.read().split()
+
+def classify_image(image):
+    pred,idx,probs = learner.predict()
+    return dict(zip(categories,map(float,probs)))
+
+ins = gr.inputs.Image()
+outs = gr.outputs.Label()
+interface = gr.Interface(fn=classify_image,inputs=ins,outputs=outs)
+interface.launch()