Update app.py

app.py

@@ -53,7 +53,7 @@ def classification(text, typology, confidence, task):
 def masking(text, task):
     model = return_model(task)
     text += ', [MASK] [MASK]'
-    pred = mask_filler(text, top_k=1)
+    pred = model(text, top_k=1)
     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