diff --git a/visual_model_test/visual_model_test.py b/visual_model_test/visual_model_test.py
index dbfc86f..bdaed07 100644
--- a/visual_model_test/visual_model_test.py
+++ b/visual_model_test/visual_model_test.py
@@ -75,12 +75,12 @@ def main(model_type, pic_name=None):
         schema = ["患者姓名", "入院日期", "出院日期", "费用总额", "个人现金支付", "个人账户支付", "自费金额", "医保类型"]
     elif model_type == "discharge":
         task_path = "../config/model/discharge_record_model"
-        test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240401000003_3_001938_2.jpg"
-        schema = ["医院", "科别", "姓名", "入院日期", "出院日期", "主治医生"]
+        test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240428000832_1_093844_2.jpg"
+        schema = ["医院", "科室", "患者姓名", "入院日期", "出院日期", "主治医生"]
     elif model_type == "cost":
         task_path = "../config/model/cost_list_model"
         test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240511000648_4_094542_2.jpg"
-        schema = ["姓名", "入院日期", "出院日期", "费用总额"]
+        schema = ["患者姓名", "入院日期", "出院日期", "费用总额"]
     elif model_type == "cost_detail":
         task_path = "../config/model/cost_list_detail_model"
         test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240511000648_4_094542_2.jpg"