Update app.py
Browse files
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):
|