from llm_module import get_llm
from ui_module import build_demo

get_llm(model_size="4b")
demo = build_demo(enable_reload=True)

if __name__ == "__main__":
    demo.launch(
        share=True, 
        server_name="0.0.0.0", 
        show_error=True
    )