Update app.py

app.py

@@ -131,13 +131,13 @@ def masking(text):
     # Case for the last position
     masked_text = ', '.join(text.split(', ')) + ', [MASK]'
     i = 0
-        while True:
-            prediction = mask_model(masked_text)[i]
-            species = prediction['token_str']
-            if species in text.split(', '):
-                i+=1
-            else:
-                break
+    while True:
+        prediction = mask_model(masked_text)[i]
+        species = prediction['token_str']
+        if species in text.split(', '):
+            i+=1
+        else:
+            break
     score = prediction['score']
     sentence = prediction['sequence']