new test torch

Dockerfile

@@ -1,12 +1,11 @@
-FROM python:3.10
+FROM python:3.9
 
 WORKDIR /code
 
 #COPY ./requirements.txt /code/requirements.txt
 
 #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
-RUN pip install torch==1.9.1 -f https://download.pytorch.org/whl/torch_stable.ht
-RUN pip install torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
+RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
 RUN pip install --no-cache-dir --upgrade -U openmim
 RUN mim install --no-cache-dir --upgrade mmengine
 RUN mim install "mmcv>=2.0.1"

README.md

@@ -5,6 +5,7 @@ colorFrom: blue
 colorTo: red
 sdk: docker
 pinned: false
+python_version: 3.9
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference