Spaces:

Disguised
/

anime_character_recognizer

Runtime error

Disguised commited on Jan 20, 2023

Commit

685c77f

1 Parent(s): 2f02313

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,13 @@
 import gradio as gr
 from fastai.vision.all import *
 import gradio as gr
 from glob import glob
 learn = load_learner('model_ft15(extra).pkl')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))

 import gradio as gr
 from fastai.vision.all import *
 import gradio as gr
+import re
 from glob import glob
 learn = load_learner('model_ft15(extra).pkl')
+categories = learn.dls.vocab
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))