Kaushik Bar commited on
Commit
1fd8303
·
1 Parent(s): 20062d7

first commit

Browse files
Files changed (1) hide show
  1. app.py +4 -5
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
- print(preds)
124
- predicted_labels = preds['labels']
125
- predicted_scores = preds['scores']
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 clean_output
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",