Update app.py

app.py

@@ -63,7 +63,7 @@ def classification(text, typology, confidence):
 
 def masking(text):
     masked_text = text + ', [MASK] [MASK]'
-    pred = fill_model(masked_text, top_k=1)
+    pred = mask_model(masked_text, top_k=1)
     new_species = [pred[i][0]['token_str'] for i in range(len(pred))]
     new_species = ' '.join(new_species)
     text = f"The last species from this vegetation plot is probably {new_species}."