write logs to separate dir
Browse files- Dockerfile +1 -0
- serve/gradio_web_server.py +3 -1
Dockerfile
CHANGED
@@ -31,5 +31,6 @@ RUN pip install -e .
|
|
31 |
RUN echo "HF_HOME (before setting it)= $HF_HOME "
|
32 |
|
33 |
ENV HF_HOME=/data/.huggingface
|
|
|
34 |
|
35 |
CMD ["sh", "/code/startup.sh"]
|
|
|
31 |
RUN echo "HF_HOME (before setting it)= $HF_HOME "
|
32 |
|
33 |
ENV HF_HOME=/data/.huggingface
|
34 |
+
RUN mkfir /logs
|
35 |
|
36 |
CMD ["sh", "/code/startup.sh"]
|
serve/gradio_web_server.py
CHANGED
@@ -17,12 +17,14 @@ import time
|
|
17 |
|
18 |
import gradio as gr
|
19 |
import requests
|
20 |
-
from llava.constants import LOGDIR
|
21 |
from llava.conversation import conv_templates, default_conversation
|
22 |
from llava.utils import build_logger, moderation_msg, server_error_msg, violates_moderation
|
23 |
|
24 |
from serve import INTERACTION_MODES_MAP, MODEL_ID_TO_NAME
|
25 |
|
|
|
|
|
26 |
logger = build_logger("gradio_web_server", "gradio_web_server.log")
|
27 |
|
28 |
headers = {"User-Agent": "PrismaticVLMs Client"}
|
|
|
17 |
|
18 |
import gradio as gr
|
19 |
import requests
|
20 |
+
# from llava.constants import LOGDIR
|
21 |
from llava.conversation import conv_templates, default_conversation
|
22 |
from llava.utils import build_logger, moderation_msg, server_error_msg, violates_moderation
|
23 |
|
24 |
from serve import INTERACTION_MODES_MAP, MODEL_ID_TO_NAME
|
25 |
|
26 |
+
LOGDIR = "/logs"
|
27 |
+
|
28 |
logger = build_logger("gradio_web_server", "gradio_web_server.log")
|
29 |
|
30 |
headers = {"User-Agent": "PrismaticVLMs Client"}
|