Update Dockerfile

Dockerfile

@@ -14,7 +14,8 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install \
     apt-get install nvidia-container-runtime
-
+    
+ENV FORCE_CUDA="1"
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
@@ -35,9 +36,7 @@ RUN pyenv install $PYTHON_VERSION && \
     pyenv global $PYTHON_VERSION && \
     pyenv rehash && \
     pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir \
-    datasets \
-    huggingface-hub "protobuf<4" "click<8.1"
+    pip install --no-cache-dir
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 # PyTorch installation with CUDA 12.1 support