Upload 9 files
Browse files- app.py +35 -7
- bart.jpg +0 -0
- bart.jpgZone.Identifier +4 -0
- ednar.jpg +0 -0
- ednar.jpgZone.Identifier +4 -0
- maggie.jpg +0 -0
- maggie.jpgZone.Identifier +4 -0
- model.pkl +3 -0
- model.pklZone.Identifier +3 -0
app.py
CHANGED
|
@@ -1,7 +1,35 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from fastai.vision.all import *
|
| 2 |
+
import gradio as gr
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
learn = load_learner('model.pkl')
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
categories = (['abraham_grampa_simpson', 'agnes_skinner', 'apu_nahasapeemapetilon', 'barney_gumble', 'bart_simpson',
|
| 10 |
+
'carl_carlson', 'charles_montgomery_burns', 'chief_wiggum', 'cletus_spuckler', 'comic_book_guy', 'disco_stu',
|
| 11 |
+
'edna_krabappel', 'fat_tony', 'gil', 'groundskeeper_willie', 'homer_simpson', 'kent_brockman', 'krusty_the_clown',
|
| 12 |
+
'lenny_leonard', 'lionel_hutz', 'lisa_simpson', 'maggie_simpson', 'marge_simpson', 'martin_prince', 'mayor_quimby',
|
| 13 |
+
'milhouse_van_houten', 'miss_hoover', 'moe_szyslak', 'ned_flanders', 'nelson_muntz', 'otto_mann', 'patty_bouvier',
|
| 14 |
+
'principal_skinner', 'professor_john_frink', 'rainier_wolfcastle', 'ralph_wiggum', 'selma_bouvier', 'sideshow_bob',
|
| 15 |
+
'sideshow_mel', 'snake_jailbird', 'troy_mcclure', 'waylon_smithers'])
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
def classify_image(img):
|
| 20 |
+
pred,idx,probs = learn.predict(img)
|
| 21 |
+
return dict(zip(categories, map(float,probs)))
|
| 22 |
+
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
image = gr.inputs.Image(shape=(192,192))
|
| 26 |
+
label = gr.outputs.Label()
|
| 27 |
+
examples = ['ednar.jpg', 'maggie.jpg', 'bart.jpg']
|
| 28 |
+
|
| 29 |
+
intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
|
| 30 |
+
intf.launch(inline=False)
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
|
| 35 |
+
|
bart.jpg
ADDED
|
bart.jpgZone.Identifier
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[ZoneTransfer]
|
| 2 |
+
ZoneId=3
|
| 3 |
+
ReferrerUrl=https://www.bing.com/
|
| 4 |
+
HostUrl=https://images.complex.com/complex/image/upload/c_limit,dpr_auto,q_90,w_720/fl_lossy,pg_1/bart_opt_wodqc5.jpg
|
ednar.jpg
ADDED
|
ednar.jpgZone.Identifier
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[ZoneTransfer]
|
| 2 |
+
ZoneId=3
|
| 3 |
+
ReferrerUrl=https://www.bing.com/
|
| 4 |
+
HostUrl=https://www.bing.com/
|
maggie.jpg
ADDED
|
maggie.jpgZone.Identifier
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[ZoneTransfer]
|
| 2 |
+
ZoneId=3
|
| 3 |
+
ReferrerUrl=https://www.bing.com/
|
| 4 |
+
HostUrl=https://th.bing.com/th/id/OIP.oChm7qTslXgG2Bu92_Y0ogHaEJ?w=303&h=180&c=7&r=0&o=5&dpr=1.5&pid=1.7
|
model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7da878764b38f4ef328833188e0947c6957ffcf426caa80038a20ec80804a20f
|
| 3 |
+
size 88159151
|
model.pklZone.Identifier
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[ZoneTransfer]
|
| 2 |
+
ZoneId=3
|
| 3 |
+
HostUrl=https://www.kaggle.com/
|