File size: 553 Bytes
6e9a95e
 
 
 
 
 
46e183a
6e9a95e
 
 
 
6efb911
 
 
46e183a
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import os

UI_VARIANT = os.getenv("MEGUMIN_UI", "default").strip().lower()

if UI_VARIANT == "messenger":
    from app_gradio_messenger import CUSTOM_CSS
    from app_gradio_messenger import MESSENGER_ALLOWED_PATHS
    from app_gradio_messenger import demo
else:
    from app_gradio import CUSTOM_CSS
    from app_gradio import demo


if __name__ == "__main__":
    launch_kwargs = {"css": CUSTOM_CSS, "ssr_mode": False}
    if UI_VARIANT == "messenger":
        launch_kwargs["allowed_paths"] = MESSENGER_ALLOWED_PATHS
    demo.launch(**launch_kwargs)