Spaces:

curt-tigges
/

anime-image-labeller

Runtime error

ctigges commited on Apr 4, 2022

Commit

ec46c47

•

1 Parent(s): a844421

Added new models trained on 300k image set

Files changed (2) hide show

anime_image_label_inference.py CHANGED Viewed

@@ -32,7 +32,7 @@ def get_pred_classes(obj, learn, thresh):
 def get_x(r): return 'images'/r['img_name']
 def get_y(r): return [t for t in r['tags'].split(' ') if t in pop_tags]
-learn = load_learner('model-large-40e.pkl')
 def predict_single_img(imf, thresh=0.2, learn=learn):

 def get_x(r): return 'images'/r['img_name']
 def get_y(r): return [t for t in r['tags'].split(' ') if t in pop_tags]
+learn = load_learner('model-large-basic-10e.pkl')
 def predict_single_img(imf, thresh=0.2, learn=learn):

model-large-basic-10e.pkl ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:a4f228d5d55434feb36678d470f61f7f8b1739e4c00ee812c9994ef09b698e2d
+size 121517173