File size: 1,515 Bytes
ddf7e50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from fastai.vision.all import *
import gradio as gr
learn = load_learner("model_2.pkl")
categories = learn.dls.vocab
for index, category in enumerate(categories):
if category == "Random Anime Photos":
categories[index] = "Others"
def classify_image(img):
pred, idx, probs = learn.predict(img)
return dict(zip(categories, map(float, probs)))
image = gr.inputs.Image(shape=(360, 360))
label = gr.outputs.Label()
examples = [
"Luffy.jpg",
"Naruto-Kurama-Mode.png",
"Goku.jpg",
"Ichigo.jpeg",
"Robin.jpeg"
]
title = "Top 10 Shounen Anime Protagonists Classifier"
description = "Fine tuned a resnet152 image classifier such that it is able to recognize protagonists of top 10 Shounen Animes.<br>"
article = """
Animes and its protagonists this image classifier will recognize:
1. One Piece - Monkey D. Luffy
2. Naruto: Shippuden - Naruto Uzumaki
3. My Hero Academia - Izuku Midoriya
4. Dragon Ball Z - Son Goku aka Kakarot
5. Fairy Tail - Natsu Dragneel
6. Yu Yu Hakusho - Yusuke Urameshi
7. Bleach - Ichigo Kurosaki
8. Hunter X Hunter - Gon Freecss
9. Fullmetal Alchemist - Edward Elric
10. Attack on Titan - Eren Yeager
Rest all other anime characters will be classified as others.
"""
intf = gr.Interface(
fn=classify_image,
inputs=image,
outputs=label,
examples=examples,
title=title,
description=description,
article=article,
theme="huggingface",
)
intf.launch(inline=False)
|