Update app.py

app.py

@@ -76,7 +76,7 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
         logits_parent = model(torch.tensor([input_ids_parent]).to(device), idx_msaseq_parent)
 
         cls_mask_logits_parent = logits_parent[0, 1, :]
-        cls_probability_parent, cls_mask_probs_parent = torch.topk((torch.softmax(cls_mask_logits_parent, dim=-1)), k=53)
+        cls_probability_parent, cls_mask_probs_parent = torch.topk((torch.softmax(cls_mask_logits_parent, dim=-1)), k=85)
 
         seqseq_parent[2] = "[MASK]"
         input_ids_parent = vocab_mlm.__getitem__(seqseq_parent)