initial commit

README.md

@@ -1,8 +1,8 @@
 ---
 title: Pokemon Detector
-emoji: 🏢
+emoji: 🐢
 colorFrom: red
-colorTo: red
+colorTo: blue
 sdk: gradio
 sdk_version: 3.4.1
 app_file: app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import os
+import fastbook
+from pathlib import Path
+from fastai.vision.widgets import *
+
+fastbook.setup_book()
+
+examples = ['./examples/pikachu.webp', './examples/charizard.webp',
+            './examples/mewtwo.jpg', './examples/rayquaza.jpeg', './examples/cinderace.webp']
+
+path = Path()
+path.ls(file_exts='.pkl')
+learn_inf = fastbook.load_learner(path/'pokemon-detector.pkl')
+
+labels = learn_inf.dls.vocab
+
+def pokemon_classifier(image):
+    image = fastbook.PILImage.create(image)
+    pred, pred_id, probs = learn_inf.predict(image)
+    output = {labels[i]: float(probs[i]) for i in range(len(labels))}
+    # limit the output to the top 5 results
+    output = dict(sorted(output.items(), key=lambda item: item[1], reverse=True)[:5])
+    return output
+
+
+iface = gr.Interface(fn=pokemon_classifier, inputs="image", outputs="label", examples=examples)
+iface.launch()

pokemon-detector.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56036eeeb6201a7f24fcbee2f5de0c2140ca782ec7dfb6647456780e78cdbb4f
+size 51544097

requirements.txt

@@ -0,0 +1,4 @@
+fastbook
+gradio
+pathlib
+fastai