Spaces:
Sleeping
Sleeping
changed path to model pickle file
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from fastai.learner import load_learner
|
|
| 4 |
|
| 5 |
def image_classifier(inp):
|
| 6 |
|
| 7 |
-
art_classifier = load_learner('
|
| 8 |
class_lst = art_classifier.dls.vocab
|
| 9 |
pred,pred_idx,probs = art_classifier.predict(inp)
|
| 10 |
pred_dict = {x:y.numpy() for x, y in zip(class_lst,probs)}
|
|
|
|
| 4 |
|
| 5 |
def image_classifier(inp):
|
| 6 |
|
| 7 |
+
art_classifier = load_learner('art_classifier.pkl')
|
| 8 |
class_lst = art_classifier.dls.vocab
|
| 9 |
pred,pred_idx,probs = art_classifier.predict(inp)
|
| 10 |
pred_dict = {x:y.numpy() for x, y in zip(class_lst,probs)}
|