File size: 672 Bytes
aa0a596
e23b947
dbd8152
aa0a596
e988c9a
1aff2c9
 
aa0a596
 
 
e23b947
 
dbd8152
 
 
 
 
 
 
 
1aff2c9
f8f08d2
1aff2c9
 
 
12a2f7f
f8f08d2
332a896
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
import logging
import os
import threading

from src.core.config import settings
from src.core.dependencies import get_answer_generator
from src.frontend.app import build_app

logger = logging.getLogger(__name__)

_demo = build_app()

if settings.LLM_EAGER_LOAD:
    def _warmup():
        try:
            get_answer_generator().load()
        except Exception as exc:
            logger.error("LLM warmup failed: %s", exc)

    threading.Thread(target=_warmup, daemon=True, name="llm-warmup").start()

if __name__ == "__main__":
    _demo.launch(
        server_name="0.0.0.0",
        server_port=settings.API_PORT,
        ssr_mode=False,
        show_error=True,
    )