nodes/modelscope-scepter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update v0.0.2

Browse Source
This commit is contained in:
jiangzeyinzi
2024-01-19 14:25:10 +08:00
parent 47c528360d
commit c038beef2b
13 changed files with 63 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scepter/studio/preprocess/caption_editor_ui/create_dataset_ui.py
+20 -9
View File
@@ -410,7 +410,9 @@ class CreateDatasetUI(UIBase):
if len(self.dataset_list) > 0 else '',
choices=self.dataset_list)
self.refresh_dataset_name.click(refresh, outputs=[self.dataset_name])
self.refresh_dataset_name.click(refresh,
outputs=[self.dataset_name],
queue=False)
def confirm_create_dataset(user_name, create_mode, file_url, file_path,
panel_state):
@@ -491,25 +493,31 @@ class CreateDatasetUI(UIBase):
self.btn_create_datasets.click(show_dataset_panel, [], [
self.file_panel, self.dataset_panel, self.btn_panel,
self.panel_state, self.user_data_name, self.create_mode
])
],
queue=False)
self.btn_create_datasets_from_file.click(show_file_panel, [], [
self.file_panel, self.dataset_panel, self.btn_panel,
self.panel_state, self.user_data_name, self.file_path,
self.file_path_url, self.create_mode
])
],
queue=False)
# Click Refresh
self.random_data_button.click(get_random_dataset_name, [],
[self.user_data_name])
[self.user_data_name],
queue=False)
self.file_path.clear(clear_file, outputs=[self.file_path_url])
self.file_path.clear(clear_file,
outputs=[self.file_path_url],
queue=False)
# Click Confirm
self.confirm_data_button.click(confirm_create_dataset, [
self.user_data_name, self.create_mode, self.file_path_url,
self.file_path, self.panel_state
], [self.panel_state, self.dataset_name])
], [self.panel_state, self.dataset_name],
queue=False)
def show_edit_panel(panel_state, data_name):
if panel_state:
@@ -527,7 +535,8 @@ class CreateDatasetUI(UIBase):
gallery_dataset.gallery_panel, gallery_dataset.upload_panel,
export_dataset.export_panel, self.modify_panel,
self.file_panel, self.btn_panel, self.user_data_name_state
])
],
queue=False)
def modify_data_name(user_name, prev_data_name):
print(
@@ -583,7 +592,8 @@ class CreateDatasetUI(UIBase):
self.modify_data_button.click(
modify_data_name,
inputs=[self.user_data_name, self.user_data_name_state],
outputs=[self.user_data_name_state, self.dataset_name])
outputs=[self.user_data_name_state, self.dataset_name],
queue=False)
def dataset_change(user_name):
if user_name is None or user_name == '':
@@ -602,4 +612,5 @@ class CreateDatasetUI(UIBase):
self.dataset_panel, self.file_panel,
self.panel_state, self.user_data_name,
gallery_dataset.gallery_state
])
],
queue=False)
scepter/studio/preprocess/caption_editor_ui/dataset_gallery_ui.py
+14 -7
View File
@@ -116,7 +116,8 @@ class DatasetGalleryUI(UIBase):
self.gallery_state.change(change_gallery,
inputs=[create_dataset.user_data_name],
outputs=[self.gl_dataset_images])
outputs=[self.gl_dataset_images],
queue=False)
def select_image(dataset_name, evt: gr.SelectData):
meta_data = create_dataset.meta_dict[dataset_name]
@@ -167,16 +168,19 @@ class DatasetGalleryUI(UIBase):
self.gl_dataset_images,
self.ori_caption, self.edit_caption,
self.info
])
],
queue=False)
self.gl_dataset_images.change(
change_image,
inputs=[create_dataset.user_data_name],
outputs=[self.ori_caption, self.edit_caption, self.info])
outputs=[self.ori_caption, self.edit_caption, self.info],
queue=False)
self.modify_button.click(
change_caption,
inputs=[create_dataset.user_data_name, self.edit_caption],
outputs=[self.ori_caption])
outputs=[self.ori_caption],
queue=False)
def delete_file(dataset_name):
cursor = create_dataset.meta_dict[dataset_name]['cursor']
@@ -218,7 +222,8 @@ class DatasetGalleryUI(UIBase):
outputs=[
self.gl_dataset_images, self.ori_caption,
self.edit_caption, self.info
])
],
queue=False)
def add_file(dataset_name, upload_image, caption):
if 'image' in upload_image:
@@ -275,7 +280,8 @@ class DatasetGalleryUI(UIBase):
outputs=[
self.gl_dataset_images, self.ori_caption,
self.edit_caption, self.info
])
],
queue=False)
def edit_caption_change(dataset_name, edit_caption):
meta = create_dataset.meta_dict[dataset_name]
@@ -285,4 +291,5 @@ class DatasetGalleryUI(UIBase):
self.edit_caption.change(
edit_caption_change,
inputs=[create_dataset.user_data_name, self.edit_caption])
inputs=[create_dataset.user_data_name, self.edit_caption],
queue=False)
scepter/studio/preprocess/caption_editor_ui/export_dataset_ui.py
+4 -2
View File
@@ -72,7 +72,8 @@ class ExportDatasetUI(UIBase):
self.export_to_zip.click(export_zip,
inputs=[create_dataset.user_data_name],
outputs=[self.export_url])
outputs=[self.export_url],
queue=False)
def export_csv(dataset_name):
meta = create_dataset.meta_dict[dataset_name]
@@ -136,4 +137,5 @@ class ExportDatasetUI(UIBase):
self.go_to_train.click(
go_to_train,
inputs=[create_dataset.user_data_name],
outputs=[manager.tabs, manager.self_train.trainer_ui.ms_data_name])
outputs=[manager.tabs, manager.self_train.trainer_ui.ms_data_name],
queue=False)