Update app.py

app.py

@@ -85,12 +85,10 @@ def masking(text, k):
     text_split = text.split(', ')
     
     best_predictions = []
-    best_positions = []
     
     for _ in range(k):
         max_score = 0
         best_prediction = None
-        best_position = None
         best_sentence = None
 
         for i in range(len(text_split) + 1):
@@ -111,12 +109,11 @@ 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)
-        best_positions.append(best_position)
-        text_split.insert(best_position, best_prediction)
+        text_split.insert(i, 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: