Update app.py

app.py

@@ -122,11 +122,8 @@ def masking(text):
         best_prediction = species
         best_position = len(text.split(', '))
     
-    masked_text = text + ', [MASK]'
-    pred = mask_model(masked_text)[0]
-    new_species = pred['token_str']
-    text = f"The most likely missing species in position {best_position} is: {best_species}."
-    image = return_species_image(new_species)
+    text = f"The most likely missing species in position {best_position} is: {best_prediction}."
+    image = return_species_image(best_prediction)
     return text, image
 
 with gr.Blocks() as demo: