kargaranamir commited on
Commit
fe01253
1 Parent(s): 48bb59c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def fasttext_prediction_in_sklearn_format(classifier, texts, num_class):
75
  # texts = [texts]
76
 
77
  res = []
78
- labels, probabilities = classifier.predict(texts, num_class)
79
  labels = remove_label_prefix_list(labels)
80
  for label, probs, text in zip(labels, probabilities, texts):
81
  order = np.argsort(np.array(label))
 
75
  # texts = [texts]
76
 
77
  res = []
78
+ labels, probabilities = classifier.predict(texts, -1)
79
  labels = remove_label_prefix_list(labels)
80
  for label, probs, text in zip(labels, probabilities, texts):
81
  order = np.argsort(np.array(label))