Spaces:

tunglinwu
/

kaiju_classifier

Sleeping

tunglinwu commited on Apr 19, 2024

Commit

a1d66c5

1 Parent(s): b83c6b6

Update

Files changed (2) hide show

app.ipynb CHANGED Viewed

The diff for this file is too large to render. See raw diff

app.py CHANGED Viewed

@@ -11,15 +11,15 @@ import gradio as gr
 learn = load_learner('model.pkl')
 # %% ../app.ipynb 18
-categories = ['Godzilla', 'King Kong']
 def classify_kaiju(img):
     pred,idx,probs=learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 # %% ../app.ipynb 20
-image = gr.inputs.Image(shape=(192, 192))
-label = gr.outputs.Label()
-examples = ['Godzilla_1.jpg', 'KingKong_1.jpg', 'Godzilla_2.jpg', 'KingKong_2.jpg']
 intf = gr.Interface(fn=classify_kaiju, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)

 learn = load_learner('model.pkl')
 # %% ../app.ipynb 18
+categories = ['Godzilla', 'King Kong', 'Mothra', 'King Ghidorah', 'Rodan']
 def classify_kaiju(img):
     pred,idx,probs=learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 # %% ../app.ipynb 20
+image = gr.Image()
+label = gr.Label()
+examples = ['Godzilla_1.jpg', 'Godzilla_2.jpg', 'KingKong_1.jpg', 'KingKong_2.jpg', 'Mothra_1.jpg', 'Mothra_2.jpg', 'King Ghidorah_1.jpg', 'King Ghidorah_2.jpg', 'Rodan_1.jpg', 'Rodan_2.jpg', ]
 intf = gr.Interface(fn=classify_kaiju, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)