Kaushik Bar
commited on
Commit
·
1fd8303
1
Parent(s):
20062d7
first commit
Browse files
app.py
CHANGED
|
@@ -120,16 +120,15 @@ def sequence_to_classify(text, labels):
|
|
| 120 |
pred["scores"], pred["labels"] = zip(*sorted_sl)
|
| 121 |
preds.append(pred)
|
| 122 |
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
clean_output = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
|
| 127 |
print("Date:{}, Sequence:{}, Labels: {}".format(
|
| 128 |
str(datetime.datetime.now()),
|
| 129 |
text,
|
| 130 |
predicted_labels))
|
| 131 |
|
| 132 |
-
return
|
| 133 |
|
| 134 |
iface = gr.Interface(
|
| 135 |
title="Alternate Zero-shot Multi-label Multilingual NLP Classifier",
|
|
|
|
| 120 |
pred["scores"], pred["labels"] = zip(*sorted_sl)
|
| 121 |
preds.append(pred)
|
| 122 |
|
| 123 |
+
predicted_labels = list(preds['labels'])
|
| 124 |
+
predicted_scores = list(preds['scores'])
|
| 125 |
+
output = {idx: float(predicted_scores.pop(0)) for idx in predicted_labels}
|
|
|
|
| 126 |
print("Date:{}, Sequence:{}, Labels: {}".format(
|
| 127 |
str(datetime.datetime.now()),
|
| 128 |
text,
|
| 129 |
predicted_labels))
|
| 130 |
|
| 131 |
+
return output
|
| 132 |
|
| 133 |
iface = gr.Interface(
|
| 134 |
title="Alternate Zero-shot Multi-label Multilingual NLP Classifier",
|