Update app.py

app.py

@@ -109,7 +109,7 @@ def CTXGen(X1, X2, τ, g_num, length_range):
             input_ids = vocab_mlm.__getitem__(generated_seq)
             logits = model(torch.tensor([input_ids]).to(device), idx_msa)
             cls_mask_logits = logits[0, 1, :]
-            cls_probability, cls_mask_probs = torch.topk((torch.softmax(cls_mask_logits, dim=-1)), k=5)
+            cls_probability, cls_mask_probs = torch.topk((torch.softmax(cls_mask_logits, dim=-1)), k=10)
 
             generated_seq[2] = "[MASK]"
             input_ids = vocab_mlm.__getitem__(generated_seq)