From 5a427e73fc3df68ca1beedce44dedb66a798c40b Mon Sep 17 00:00:00 2001 From: Nakarin Satayasai Date: Fri, 3 Nov 2023 11:05:12 +0700 Subject: [PATCH] update --- py/io/csv_prompts_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/io/csv_prompts_loader.py b/py/io/csv_prompts_loader.py index 91b91b4..eed2b49 100644 --- a/py/io/csv_prompts_loader.py +++ b/py/io/csv_prompts_loader.py @@ -57,7 +57,7 @@ class CSVPromptsLoader(): return ("", "", "", "", "",) df = pd.read_csv(path, index_col=0, dtype={ - "name": 'string', "styles": 'string', "face": 'string', "cloth": 'string', "action": 'string', "face_action": 'string'}) + "name": 'string', "styles": 'string', "face": 'string', "cloth": 'string', "action": 'string', "face_action": 'string', "environment": 'string'}) if remove_extension == "On": key = os.path.splitext(key)[0] @@ -68,4 +68,4 @@ class CSVPromptsLoader(): logger.exception("Key is not exist") return ("", "", "", "", "",) - return (values['styles'], values['face'], values['cloth'], values['action'], values['face_action'],) + return (values['styles'], values['face'], values['cloth'], values['action'], values['face_action'], values['environment'])