Upload 3 files

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from fastai.vision.all import *
+
+# Loading the model we trained
+learn = load_learner('one_piece.pkl')
+
+# Prediction function
+def predict_character(img):
+    pred_class, pred_idx, probs = learn.predict(img)
+
+    return {learn.dls.vocab[i]: float(probs[i])
+            for i in range(len(learn.dls.vocab))}
+
+# Gradio app interface
+demo = gr.Interface(
+    fn = predict_character,
+    inputs = gr.Image(type = 'pil', label = 'Upload Character Image'),
+    outputs = gr.Label(num_top_classes = 10, label = 'Predictions'),
+    title = 'One Piece Straw Hat Recognition',
+    description = 'Upload any straw hats members image except franky and jinbe',
+    examples = None,
+    theme = gr.themes.Soft()
+)
+
+# Launch our app
+demo.launch()

one_piece.pkl

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