Added model and interface

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
+from fastai.vision.all import load_learner
+from PIL import Image
 
-def greet(name):
-    return "Hello " + name + "!!"
+model = load_learner('./export.pkl')  
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def classify_image(img):
+    # Convert the image to a format the model expects
+    img = Image.fromarray(img.astype('uint8'), 'RGB')
+    # Make a prediction
+    pred, idx, probs = model.predict(img)
+    # Return the result
+    return {model.dls.vocab[i]: float(probs[i]) for i in range(len(model.dls.vocab))}
+
+demo = gr.Interface(fn=classify_image, inputs="image", outputs="label")
+
+if __name__ == "__main__":
+    demo.launch()

export.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21d03b139e8d4f7e07eebfe46348fdc1306f89e23b82eaa5a54791ab8007f548
+size 46971454