mattb512 commited on
Commit
0eb9eab
1 Parent(s): 3a5b37d

write logs to separate dir

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. 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"}