diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py new file mode 100644 index 00000000..389bc1e2 --- /dev/null +++ b/modules/ui_image_generation.py @@ -0,0 +1,9 @@ +import gradio as gr + + +def create_ui(): + pass + + +def create_event_handlers(): + pass diff --git a/server.py b/server.py index c804c342..87bbdc4a 100644 --- a/server.py +++ b/server.py @@ -50,6 +50,7 @@ from modules import ( ui_chat, ui_default, ui_file_saving, + ui_image_generation, ui_model_menu, ui_notebook, ui_parameters, @@ -163,6 +164,7 @@ def create_interface(): ui_chat.create_character_settings_ui() # Character tab ui_model_menu.create_ui() # Model tab if not shared.args.portable: + ui_image_generation.create_ui() # Image generation tab training.create_ui() # Training tab ui_session.create_ui() # Session tab