Ywung commited on
Commit
e2b1957
1 Parent(s): 1c6a7e8

try to fix datasets dependency

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -0
  2. README.md +5 -5
  3. requirements.txt +1 -1
Dockerfile CHANGED
@@ -13,6 +13,7 @@ RUN python3 -m venv /build/venv
13
  RUN . /build/venv/bin/activate && \
14
  pip3 install --upgrade pip setuptools wheel && \
15
  pip3 install torch torchvision torchaudio && \
 
16
  pip3 install -r requirements.txt
17
 
18
  # https://developer.nvidia.com/cuda-gpus
 
13
  RUN . /build/venv/bin/activate && \
14
  pip3 install --upgrade pip setuptools wheel && \
15
  pip3 install torch torchvision torchaudio && \
16
+ pip3 install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1" "pydantic~=1.0" && \
17
  pip3 install -r requirements.txt
18
 
19
  # https://developer.nvidia.com/cuda-gpus
README.md CHANGED
@@ -4,11 +4,11 @@ emoji: 🤖
4
  colorFrom: blue
5
  colorTo: purple
6
  pinned: true
7
- sdk: gradio
8
- sdk_version: 3.41.2
9
- app_file: server.py
10
- # sdk: docker
11
- # app_port: 7860
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
4
  colorFrom: blue
5
  colorTo: purple
6
  pinned: true
7
+ # sdk: gradio
8
+ # sdk_version: 3.41.2
9
+ # app_file: server.py
10
+ sdk: docker
11
+ app_port: 7860
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
requirements.txt CHANGED
@@ -23,7 +23,7 @@ tqdm
23
  wandb
24
 
25
  # datasets
26
- https://files.pythonhosted.org/packages/fe/17/5825fdf034ff1a315becdbb9b6fe5a2bd9d8e724464535f18809593bf9c2/datasets-2.10.1-py3-none-any.whl; platform_system == "Windows"
27
 
28
  # bitsandbytes
29
  bitsandbytes==0.41.1; platform_system != "Windows"
 
23
  wandb
24
 
25
  # datasets
26
+ https://files.pythonhosted.org/packages/fe/17/5825fdf034ff1a315becdbb9b6fe5a2bd9d8e724464535f18809593bf9c2/datasets-2.10.1-py3-none-any.whl; platform_system != "Windows"
27
 
28
  # bitsandbytes
29
  bitsandbytes==0.41.1; platform_system != "Windows"