oucgc1996 commited on
Commit
59040e5
·
verified ·
1 Parent(s): adf7ae1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -121,6 +121,7 @@ def CTXGen(τ, g_num, length_range, model_name):
121
  cls_probability = cls_probability[0].item()
122
  act_probability = act_probability[0].item()
123
  generated_seq = generated_seq[generated_seq.index('[MASK]') + 2:generated_seq.index('[SEP]')]
 
124
  if generated_seq.count('C') % 2 == 0 and len("".join(generated_seq)) == gen_len:
125
  generated_seqs.append("".join(generated_seq))
126
  if "".join(generated_seq) not in train_seq and "".join(generated_seq) not in generated_seqs[0:-1] and all(x not in NON_AA for x in generated_seq):
 
121
  cls_probability = cls_probability[0].item()
122
  act_probability = act_probability[0].item()
123
  generated_seq = generated_seq[generated_seq.index('[MASK]') + 2:generated_seq.index('[SEP]')]
124
+ print(generated_seq)
125
  if generated_seq.count('C') % 2 == 0 and len("".join(generated_seq)) == gen_len:
126
  generated_seqs.append("".join(generated_seq))
127
  if "".join(generated_seq) not in train_seq and "".join(generated_seq) not in generated_seqs[0:-1] and all(x not in NON_AA for x in generated_seq):