Spaces:
Build error
Build error
File size: 2,609 Bytes
f889344 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import gradio as gr
from .class_gui_config import KohyaSSGUIConfig
class MetaData:
def __init__(
self,
config: KohyaSSGUIConfig = {},
) -> None:
self.config = config
with gr.Row():
self.metadata_title = gr.Textbox(
label="Metadata title",
placeholder="(optional) title for model metadata (default is output_name)",
interactive=True,
value=self.config.get("metadata.title", ""),
)
self.metadata_author = gr.Textbox(
label="Metadata author",
placeholder="(optional) author name for model metadata",
interactive=True,
value=self.config.get("metadata.author", ""),
)
self.metadata_description = gr.Textbox(
label="Metadata description",
placeholder="(optional) description for model metadata",
interactive=True,
value=self.config.get("metadata.description", ""),
)
with gr.Row():
self.metadata_license = gr.Textbox(
label="Metadata license",
placeholder="(optional) license for model metadata",
interactive=True,
value=self.config.get("metadata.license", ""),
)
self.metadata_tags = gr.Textbox(
label="Metadata tags",
placeholder="(optional) tags for model metadata, separated by comma",
interactive=True,
value=self.config.get("metadata.tags", ""),
)
def run_cmd(run_cmd: list, **kwargs):
if "metadata_title" in kwargs and kwargs.get("metadata_title") != "":
run_cmd.append("--metadata_title")
run_cmd.append(kwargs["metadata_title"])
if "metadata_author" in kwargs and kwargs.get("metadata_author") != "":
run_cmd.append("--metadata_author")
run_cmd.append(kwargs["metadata_author"])
if "metadata_description" in kwargs and kwargs.get("metadata_description") != "":
run_cmd.append("--metadata_description")
run_cmd.append(kwargs["metadata_description"])
if "metadata_license" in kwargs and kwargs.get("metadata_license") != "":
run_cmd.append("--metadata_license")
run_cmd.append(kwargs["metadata_license"])
if "metadata_tags" in kwargs and kwargs.get("metadata_tags") != "":
run_cmd.append("--metadata_tags")
run_cmd.append(kwargs["metadata_tags"])
return run_cmd
|