Spaces:
Running
Running
Update app.py
Browse files
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,
|
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))
|