File size: 666 Bytes
92cd965
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
# Merge block weighted Board
#
# extension of AUTOMATIC1111 web ui
#
# 2022/12/14 bbc_mc
#

import os
import gradio as gr

from modules import script_callbacks


from scripts.mbw import ui_mbw
from scripts.mbw_each import ui_mbw_each


#
# UI callback
#
def on_ui_tabs():

    with gr.Blocks() as main_block:
        with gr.Tab("MBW", elem_id="tab_mbw"):
            ui_mbw.on_ui_tabs()

        with gr.Tab("MBW Each", elem_id="tab_mbw_each"):
            ui_mbw_each.on_ui_tabs()

    # return required as (gradio_component, title, elem_id)
    return (main_block, "Merge Block Weighted", "merge_block_weighted"),

# on_UI
script_callbacks.on_ui_tabs(on_ui_tabs)