Update app.py

app.py

@@ -44,7 +44,8 @@ def get_all_predictions(text_sentence, top_clean=5):
   with torch.no_grad():
     predict = bert_model(input_ids)[0]
   bert = decode(bert_tokenizer, predict[0, mask_idx, :].topk(top_k).indices.tolist(), top_clean)
-  return {'bert': bert}
+  cls = decode(bert_tokenizer, predict[0, 0, :].topk(top_k).indices.tolist(), top_clean)
+  return {'bert': bert,'[CLS]':cls}
 
 def get_bert_prediction(input_text,top_k):
   try: