Spaces:
Runtime error
Runtime error
Updated app to use new model
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def get_pred_classes(obj, learn, thresh):
|
|
| 32 |
def get_x(r): return 'images'/r['img_name']
|
| 33 |
def get_y(r): return [t for t in r['tags'].split(' ') if t in pop_tags]
|
| 34 |
|
| 35 |
-
learn = load_learner(fname='model-large-
|
| 36 |
|
| 37 |
def predict_single_img(imf, thresh=0.2, learn=learn):
|
| 38 |
|
|
|
|
| 32 |
def get_x(r): return 'images'/r['img_name']
|
| 33 |
def get_y(r): return [t for t in r['tags'].split(' ') if t in pop_tags]
|
| 34 |
|
| 35 |
+
learn = load_learner(fname='model-large-basic-10e.pkl')
|
| 36 |
|
| 37 |
def predict_single_img(imf, thresh=0.2, learn=learn):
|
| 38 |
|