Hugging Face's logo

Spaces:
KeeganFdes
/
stack_onnx
Runtime error

App Files Community
KeeganFdes commited on
Commit
31071b3
•
1 Parent(s): b93945b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def preprocess_text(text):
36
 
37
 
38
  def predict_tags(text):
39
- return mlb[np.where(model.predict(vectorizer.transform([preprocess_text(text)])).flatten() == 1)]
40
 
41
 
42
 
 
36
 
37
 
38
  def predict_tags(text):
39
+ return mlb.classes_[np.where(model.predict(vectorizer.transform([preprocess_text(text)])).flatten() == 1)]
40
 
41
 
42