Update app.py

app.py

@@ -54,7 +54,8 @@ def masking(text, task):
     model = return_model(task)
     text += ', [MASK] [MASK]'
     pred = model(text, top_k=1)
-    text = pred[0]["sequence"]
+    text = pred["sequence"]
+    #text = pred[0]["sequence"]
     image = gr.Image(value="https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png")
     return text, image