Update seq_gen.py

seq_gen.py

@@ -31,7 +31,10 @@ def seq_score(input_cdr, gen_list, model, tar_emb, out_path):
     df['seq'] = seq_com[0:-1]
     df['score'] = scores[0:-1]
     df.sort_values(by='score', inplace=True, ascending=False)
+    seq_highest = df['seq'].tolist()[0]
+    with open('structure_seq.txt', 'w') as f:
+        f.write(seq_highest+'\n')
     input_seq = seq_com[-1]
     scores = scores[-1]
     df.loc[len(df.index)] = [input_seq, scores]
-    df.to_csv(out_path, sep=',', index=False, header=True)
+    df.to_csv(out_path, sep=',', index=False, header=True)
\ No newline at end of file