Update app.py

app.py

@@ -89,6 +89,7 @@ def masking(text, k):
     for _ in range(k):
         max_score = 0
         best_prediction = None
+        best_position = None
         best_sentence = None
 
         for i in range(len(text_split) + 1):
@@ -109,11 +110,14 @@ def masking(text, k):
             if score > max_score:
                 max_score = score
                 best_prediction = species
+                best_position = i
                 best_sentence = sentence
         
         best_predictions.append(best_prediction)
-        text_split.insert(i, best_prediction)
-        best_positions = [text_split.index(prediction) for prediction in best_predictions]
+        text_split.insert(best_position, best_prediction)
+        
+    best_positions = [text_split.index(prediction) for prediction in best_predictions]
+    
     if k == 1:
         text = f"The most likely missing species is {best_predictions[0]} (position {best_positions[0]})."
     else: