Megumin-chat / app.py
Junhoee's picture
Update app.py
46e183a verified
raw
history blame contribute delete
553 Bytes
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)