Spaces:

tunglinwu
/

kaiju_classifier

Sleeping

tunglinwu commited on Apr 19, 2024

Commit

2c34937

verified ·

1 Parent(s): 57b04fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import gradio as gr
 learn = load_learner('model.pkl')
 # %% ../app.ipynb 18
-categories = ['Godzilla', 'King Ghidorah', 'Mothra', 'King Kong', 'Rodan']
 def classify_kaiju(img):
     pred,idx,probs=learn.predict(img)
     return dict(zip(categories, map(float,probs)))

 learn = load_learner('model.pkl')
 # %% ../app.ipynb 18
+categories = ['Godzilla', 'King Ghidorah', 'King Kong', 'Mothra', 'Rodan']
 def classify_kaiju(img):
     pred,idx,probs=learn.predict(img)
     return dict(zip(categories, map(float,probs)))