Update app.py

app.py

@@ -64,7 +64,7 @@ def fill_masked(text: str, model_name: str, top_k: int):
                     "score": round(cand["score"], 4),
                     "token": cand["token_str"]
                 })
-            res_dict[mask_num] = formatted
+            res_dict[f"mask_{str(mask_num)}"] = formatted
     elif isinstance(results, list):
         formatted = []
         for cand in results:
@@ -73,7 +73,7 @@ def fill_masked(text: str, model_name: str, top_k: int):
                 "score": round(cand["score"], 4),
                 "token": cand["token_str"]
             })
-        res_dict[0] = formatted
+        res_dict[f"mask_{str(0)}"] = formatted
     else:
         raise ValueError("The results need to be presented in a list or a list of lists")
     return res_dict