Spaces:
Running
Running
greos
commited on
Commit
•
cb6ff85
1
Parent(s):
0cb16b7
No need to log
Browse files
app.py
CHANGED
|
@@ -5,11 +5,9 @@ import os
|
|
| 5 |
# Load the learner model
|
| 6 |
learn_inf = load_learner("model/model.pkl")
|
| 7 |
|
|
|
|
| 8 |
categories = learn_inf.dls.vocab
|
| 9 |
|
| 10 |
-
# Log the categories
|
| 11 |
-
print(categories)
|
| 12 |
-
|
| 13 |
# Comma separated string of categories with and between the last two
|
| 14 |
categories_str = ", ".join(categories[:-1]) + " or " + categories[-1]
|
| 15 |
|
|
|
|
| 5 |
# Load the learner model
|
| 6 |
learn_inf = load_learner("model/model.pkl")
|
| 7 |
|
| 8 |
+
# Get the categories from the model
|
| 9 |
categories = learn_inf.dls.vocab
|
| 10 |
|
|
|
|
|
|
|
|
|
|
| 11 |
# Comma separated string of categories with and between the last two
|
| 12 |
categories_str = ", ".join(categories[:-1]) + " or " + categories[-1]
|
| 13 |
|