Spaces:
Runtime error
Runtime error
Commit
·
abef4a0
1
Parent(s):
916d928
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ with open('id_to_category.pkl', 'rb') as fp:
|
|
| 33 |
id_to_category = pickle.load(fp)
|
| 34 |
|
| 35 |
def text_classifier(text):
|
| 36 |
-
predictions=end_to_end_model.predict(
|
| 37 |
for pred in predictions:
|
| 38 |
return(id_to_category[np.argmax(pred)])
|
| 39 |
|
|
|
|
| 33 |
id_to_category = pickle.load(fp)
|
| 34 |
|
| 35 |
def text_classifier(text):
|
| 36 |
+
predictions=end_to_end_model.predict(text)
|
| 37 |
for pred in predictions:
|
| 38 |
return(id_to_category[np.argmax(pred)])
|
| 39 |
|