Spaces:
Sleeping
Sleeping
Commit
·
362c3f8
1
Parent(s):
ab9a86f
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,6 +28,8 @@ logger.setLevel(logging.DEBUG)
|
|
| 28 |
|
| 29 |
with open("list.dat", 'rb') as f:
|
| 30 |
categories = pickle.load(f)
|
|
|
|
|
|
|
| 31 |
logging.debug(f"CATEGORIES: {type(categories)}")
|
| 32 |
logging.debug(f"{type(categories[0])}")
|
| 33 |
model = load_learner("model.pkl")
|
|
|
|
| 28 |
|
| 29 |
with open("list.dat", 'rb') as f:
|
| 30 |
categories = pickle.load(f)
|
| 31 |
+
categories = [x.item() for x in categories]
|
| 32 |
+
|
| 33 |
logging.debug(f"CATEGORIES: {type(categories)}")
|
| 34 |
logging.debug(f"{type(categories[0])}")
|
| 35 |
model = load_learner("model.pkl")
|