From 69ddbac993af776559ae93ed01fef3dac0980305 Mon Sep 17 00:00:00 2001
From: Jingfei Hou <houjf20@mails.tsinghua.edu.cn>
Date: Tue, 10 Oct 2023 08:51:11 +0000
Subject: [PATCH] Update demo.py

---
 demo.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/demo.py b/demo.py
index f9f978a..c96e9a7 100644
--- a/demo.py
+++ b/demo.py
@@ -25,3 +25,6 @@ if __name__ == '__main__':
     model = torch.load('./results/models/'+tar_specie+'.model.pt', map_location='cpu')
     # 进行评分并返回生成的抗体数据（完整，包括FR和CDR区）
     seq_score(input_cdr, gen_list, model, tar_emb, 'ans_com.csv')
+    print("The results of the build can be found in the ans_com file.")
+    print("The last sequence is the input sequence.")
+    print("The highest-rated sequence is stored in structure_seq file.")
-- 
GitLab