nateraw commited on
Commit
b1db9e9
·
1 Parent(s): ea62ba7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -1
Dockerfile CHANGED
@@ -120,6 +120,14 @@ ENV HUGGINGFACE_HUB_CACHE=/data \
120
  NUM_SHARD=1 \
121
  PORT=7860
122
 
 
 
 
 
 
 
 
 
123
  LABEL com.nvidia.volumes.needed="nvidia_driver"
124
 
125
  WORKDIR /usr/src
@@ -177,4 +185,6 @@ FROM base
177
 
178
  # ENTRYPOINT ["text-generation-launcher"]
179
  # CMD ["--json-output"]
180
- CMD ["text-generation-launcher", "--json-output"]
 
 
 
120
  NUM_SHARD=1 \
121
  PORT=7860
122
 
123
+ ENV PYTHONPATH=$HOME/app \
124
+ PYTHONUNBUFFERED=1 \
125
+ GRADIO_ALLOW_FLAGGING=never \
126
+ GRADIO_NUM_PORTS=1 \
127
+ GRADIO_SERVER_NAME=0.0.0.0 \
128
+ GRADIO_THEME=huggingface \
129
+ SYSTEM=spaces
130
+
131
  LABEL com.nvidia.volumes.needed="nvidia_driver"
132
 
133
  WORKDIR /usr/src
 
185
 
186
  # ENTRYPOINT ["text-generation-launcher"]
187
  # CMD ["--json-output"]
188
+ COPY app.py app.py
189
+ CMD ["python", "app.py"]
190
+ # CMD ["text-generation-launcher", "--json-output"]