Update Dockerfile

Dockerfile

@@ -8,8 +8,8 @@ ENV HOME=/home/user \
 WORKDIR $HOME/app
 
 RUN pip install -U pip setuptools wheel pandas einops ray==2.10.0 huggingface_hub safetensors
-RUN pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
-RUN pip install autogluon.tabular[all,tabpfn]==1.2.0
+RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
+RUN pip install autogluon.tabular tabpfn==2.1.0
 
 COPY --chown=user . $HOME/app
 CMD ["python", "app.py"]