Abhishek Thakur commited on
Commit
aaebb20
·
1 Parent(s): f6da064

update docker image

Browse files
Files changed (2) hide show
  1. Dockerfile +4 -0
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -46,6 +46,10 @@ ENV HOME=/app
46
 
47
  ENV PYTHONPATH=$HOME/app \
48
  PYTHONUNBUFFERED=1 \
 
 
 
 
49
  SYSTEM=spaces
50
 
51
 
 
46
 
47
  ENV PYTHONPATH=$HOME/app \
48
  PYTHONUNBUFFERED=1 \
49
+ GRADIO_ALLOW_FLAGGING=never \
50
+ GRADIO_NUM_PORTS=1 \
51
+ GRADIO_SERVER_NAME=0.0.0.0 \
52
+ GRADIO_THEME=huggingface \
53
  SYSTEM=spaces
54
 
55
 
requirements.txt CHANGED
@@ -25,7 +25,7 @@ einops==0.6.1
25
  peft==0.7.1
26
  trl==0.7.4
27
  tiktoken==0.5.1
28
- transformers==4.36.1
29
  accelerate==0.25.0
30
  diffusers==0.21.4
31
  bitsandbytes==0.41.0
 
25
  peft==0.7.1
26
  trl==0.7.4
27
  tiktoken==0.5.1
28
+ transformers==4.37.0
29
  accelerate==0.25.0
30
  diffusers==0.21.4
31
  bitsandbytes==0.41.0