Update app.py

app.py

@@ -59,9 +59,9 @@ def get_all_predictions(text_sentence, model_name,top_clean=5):
   cls = decode(bert_tokenizer, predict[0, 0, :].topk(top_k*5).indices.tolist(), top_clean)
   
   if ("[MASK]" in text_sentence or "<mask>" in text_sentence):
-    return {'Input sentence':text_sentence,'Tokenized text': tokenized_text,'Model':model_name,'Masked position': bert,'[CLS]':cls}
+    return {'Input sentence':text_sentence,'Tokenized text': tokenized_text, 'results_count':top_k,'Model':model_name,'Masked position': bert,'[CLS]':cls}
   else:
-    return {'Input sentence':text_sentence,'Tokenized text': tokenized_text,'Model':model_name,'[CLS]':cls}
+    return {'Input sentence':text_sentence,'Tokenized text': tokenized_text,'results_count':top_k,'Model':model_name,'[CLS]':cls}
 
 def get_bert_prediction(input_text,top_k,model_name):
   try: