Add app.py

.gitignore

@@ -0,0 +1 @@
+resnet18.pkl

app.py

@@ -0,0 +1,23 @@
+import gradio as gr
+from fastai.vision.all import *
+
+learn = load_learner("resnet18.pkl")
+
+categories = ("Peacock Spot", "Healthy")
+
+
+def classify_health(input_img):
+    pred, idx, probs = learn.predict(input_img)
+    return dict(zip(categories, map(float, probs)))
+
+
+labels = gr.outputs.Label()
+examples = [
+    "examples/healthy.jpg",
+    "examples/healthy2.jpg",
+    "examples/peacock_spot.jpeg",
+]
+demo = gr.Interface(
+    classify_health, gr.Image(height=224, width=224), outputs=labels, examples=examples
+)
+demo.launch(inline=False)