Update app.py

app.py

@@ -100,7 +100,7 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
             if is_stopped:
                 return pd.DataFrame(), "output.csv"
 
-            _, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, None)
+            _, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)
             idx_seq = torch.tensor(idx_seq).unsqueeze(0).to(device)
             idx_msa = torch.tensor(idx_msa).unsqueeze(0).to(device)
             attn_idx = torch.tensor(attn_idx).to(device)
@@ -115,7 +115,8 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
             predicted_token = vocab_mlm.to_tokens(int(predicted_token_id))
             input_text[mask_position.item()] = predicted_token
             padded_seq[mask_position.item()] = predicted_token.strip()
-
+            new_seq = padded_seq
+            
         generated_seq = input_text
 
         generated_seq[1] = "[MASK]"