Spaces:
Running
Running
| # Copyright 2024 the LlamaFactory team. | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| from demo.utils import csv_to_size_list | |
| from demo.config import load_configuration | |
| import os | |
| base_dir = os.path.dirname(os.path.abspath(__file__)) | |
| size_list_dict_CN = csv_to_size_list(os.path.join(base_dir, "assets/size_list_CN.csv")) | |
| size_list_dict_EN = csv_to_size_list(os.path.join(base_dir, "assets/size_list_EN.csv")) | |
| ( | |
| size_list_config_CN, | |
| size_list_config_EN, | |
| color_list_dict_CN, | |
| color_list_dict_EN, | |
| ) = load_configuration(base_dir) | |
| LOCALES = { | |
| "face_model": { | |
| "en": { | |
| "label": "Face detection model", | |
| }, | |
| "zh": { | |
| "label": "人脸检测模型", | |
| }, | |
| }, | |
| "matting_model": { | |
| "en": { | |
| "label": "Matting model", | |
| }, | |
| "zh": { | |
| "label": "抠图模型", | |
| }, | |
| }, | |
| "key_param": { | |
| "en": { | |
| "label": "Key Parameters", | |
| }, | |
| "zh": { | |
| "label": "核心参数", | |
| }, | |
| }, | |
| "advance_param": { | |
| "en": { | |
| "label": "Advance Parameters", | |
| }, | |
| "zh": { | |
| "label": "高级参数", | |
| }, | |
| }, | |
| "size_mode": { | |
| "en": { | |
| "label": "ID photo size options", | |
| "choices": ["Size List", "Only Change Background", "Custom Size"], | |
| "custom_size_eror": "The width should not be greater than the length; the length and width should not be less than 100, and no more than 1800.", | |
| }, | |
| "zh": { | |
| "label": "证件照尺寸选项", | |
| "choices": ["尺寸列表", "只换底", "自定义尺寸"], | |
| "custom_size_eror": "宽度不应大于长度；长度和宽度不应小于100，不大于1800。", | |
| }, | |
| }, | |
| "size_list": { | |
| "en": { | |
| "label": "Size list", | |
| "choices": list(size_list_dict_EN.keys()), | |
| "develop": size_list_config_EN, | |
| }, | |
| "zh": { | |
| "label": "预设尺寸", | |
| "choices": list(size_list_dict_CN.keys()), | |
| "develop": size_list_config_CN, | |
| }, | |
| }, | |
| "bg_color": { | |
| "en": { | |
| "label": "Background color", | |
| "choices": list(color_list_dict_EN.keys()) + ["Custom"], | |
| "develop": color_list_dict_EN, | |
| }, | |
| "zh": { | |
| "label": "背景颜色", | |
| "choices": list(color_list_dict_CN.keys()) + ["自定义底色"], | |
| "develop": color_list_dict_CN, | |
| }, | |
| }, | |
| "button": { | |
| "en": { | |
| "label": "Start", | |
| }, | |
| "zh": { | |
| "label": "开始制作", | |
| }, | |
| }, | |
| "head_measure_ratio": { | |
| "en": { | |
| "label": "Head ratio", | |
| }, | |
| "zh": { | |
| "label": "面部比例", | |
| }, | |
| }, | |
| "top_distance": { | |
| "en": { | |
| "label": "Top distance", | |
| }, | |
| "zh": { | |
| "label": "头距顶距离", | |
| }, | |
| }, | |
| "image_kb": { | |
| "en": { | |
| "label": "Set KB size", | |
| "choices": ["Not Set", "Custom"], | |
| }, | |
| "zh": { | |
| "label": "设置 KB 大小", | |
| "choices": ["不设置", "自定义"], | |
| }, | |
| }, | |
| "image_kb_size": { | |
| "en": { | |
| "label": "KB size", | |
| }, | |
| "zh": { | |
| "label": "KB 大小", | |
| }, | |
| }, | |
| "render_mode": { | |
| "en": { | |
| "label": "Render mode", | |
| "choices": [ | |
| "Solid Color", | |
| "Up-Down Gradient (White)", | |
| "Center Gradient (White)", | |
| ], | |
| }, | |
| "zh": { | |
| "label": "渲染方式", | |
| "choices": ["纯色", "上下渐变（白色）", "中心渐变（白色）"], | |
| }, | |
| }, | |
| # Tab3 - 水印工作台 | |
| "watermark_tab": { | |
| "en": { | |
| "label": "Watermark", | |
| }, | |
| "zh": { | |
| "label": "水印", | |
| }, | |
| }, | |
| "watermark_text": { | |
| "en": { | |
| "label": "Text", | |
| "value": "Hello", | |
| "placeholder": "up to 20 characters", | |
| }, | |
| "zh": { | |
| "label": "水印文字", | |
| "value": "Hello", | |
| "placeholder": "最多20个字符", | |
| }, | |
| }, | |
| "watermark_color": { | |
| "en": { | |
| "label": "Color", | |
| }, | |
| "zh": { | |
| "label": "水印颜色", | |
| }, | |
| }, | |
| "watermark_size": { | |
| "en": { | |
| "label": "Size", | |
| }, | |
| "zh": { | |
| "label": "文字大小", | |
| }, | |
| }, | |
| "watermark_opacity": { | |
| "en": { | |
| "label": "Opacity", | |
| }, | |
| "zh": { | |
| "label": "水印透明度", | |
| }, | |
| }, | |
| "watermark_angle": { | |
| "en": { | |
| "label": "Angle", | |
| }, | |
| "zh": { | |
| "label": "水印角度", | |
| }, | |
| }, | |
| "watermark_space": { | |
| "en": { | |
| "label": "Space", | |
| }, | |
| "zh": { | |
| "label": "水印间距", | |
| }, | |
| }, | |
| "watermark_switch": { | |
| "en": { | |
| "label": "Watermark", | |
| "value": "Not Add", | |
| "choices": ["Not Add", "Add"], | |
| }, | |
| "zh": { | |
| "label": "水印", | |
| "value": "不添加", | |
| "choices": ["不添加", "添加"], | |
| }, | |
| }, | |
| # 输出结果 | |
| "notification": { | |
| "en": { | |
| "label": "notification", | |
| "face_error": "The number of faces is not equal to 1, please upload an image with a single face. If the actual number of faces is 1, it may be an issue with the accuracy of the detection model. Please switch to a different face detection model on the left or raise a Github Issue to notify the author.", | |
| }, | |
| "zh": { | |
| "label": "通知", | |
| "face_error": "人脸数不等于1，请上传单人照片。如果实际人脸数为1，可能是检测模型的准确度问题，请切换左侧不同的人脸检测模型或提出Github Issue通知作者。", | |
| }, | |
| }, | |
| "standard_photo": { | |
| "en": { | |
| "label": "Standard photo", | |
| }, | |
| "zh": { | |
| "label": "标准照", | |
| }, | |
| }, | |
| "hd_photo": { | |
| "en": { | |
| "label": "HD photo", | |
| }, | |
| "zh": { | |
| "label": "高清照", | |
| }, | |
| }, | |
| "standard_photo_png": { | |
| "en": { | |
| "label": "Matting Standard photo", | |
| }, | |
| "zh": { | |
| "label": "透明标准照", | |
| }, | |
| }, | |
| "hd_photo_png": { | |
| "en": { | |
| "label": "Matting HD photo", | |
| }, | |
| "zh": { | |
| "label": "透明高清照", | |
| }, | |
| }, | |
| "layout_photo": { | |
| "en": { | |
| "label": "Layout photo", | |
| }, | |
| "zh": { | |
| "label": "六寸排版照", | |
| }, | |
| }, | |
| "download": { | |
| "en": { | |
| "label": "Download the photo after adjusting the KB size", | |
| }, | |
| "zh": { | |
| "label": "下载调整 KB 大小后的照片", | |
| }, | |
| }, | |
| "matting_image": { | |
| "en": { | |
| "label": "Matting image", | |
| }, | |
| "zh": { | |
| "label": "抠图图像", | |
| }, | |
| }, | |
| "beauty_tab": { | |
| "en": { | |
| "label": "Beauty", | |
| }, | |
| "zh": { | |
| "label": "美颜", | |
| }, | |
| }, | |
| "whitening_strength": { | |
| "en": { | |
| "label": "whitening strength", | |
| }, | |
| "zh": { | |
| "label": "美白强度", | |
| }, | |
| }, | |
| } | |