orpatashnik commited on
Commit
17e31af
·
1 Parent(s): f183b35
Files changed (2) hide show
  1. Dockerfile +9 -0
  2. app.py +1 -2
Dockerfile CHANGED
@@ -2,6 +2,15 @@ FROM yogi0421/dlib-base-image:python38-dlib19.24.0
2
 
3
  WORKDIR /app
4
 
 
 
 
 
 
 
 
 
 
5
  COPY requirements.txt .
6
  RUN pip install --upgrade pip \
7
  && pip install -r requirements.txt
 
2
 
3
  WORKDIR /app
4
 
5
+ # Choose a writable directory, e.g., /home/user/huggingface
6
+ RUN mkdir -p /home/user/huggingface
7
+ RUN chmod -R 777 /home/user/huggingface
8
+
9
+ # Then set environment variables
10
+ ENV HF_HOME=/home/user/huggingface \
11
+ HF_HUB_CACHE=/home/user/huggingface/hub \
12
+ TRANSFORMERS_CACHE=/home/user/huggingface/hub
13
+
14
  COPY requirements.txt .
15
  RUN pip install --upgrade pip \
16
  && pip install -r requirements.txt
app.py CHANGED
@@ -111,5 +111,4 @@ with gr.Blocks() as demo:
111
  outputs=output_gallery
112
  )
113
 
114
- if __name__ == "__main__":
115
- demo.launch(share=True, debug=True)
 
111
  outputs=output_gallery
112
  )
113
 
114
+ demo.launch()