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",
|