Update Dockerfile

Dockerfile

@@ -6,7 +6,6 @@ RUN useradd -m -u 1000 user
 
 WORKDIR /tmp_build
 COPY requirements.txt .
-
 RUN pip install --no-cache-dir --upgrade pip \
  && pip install --no-cache-dir -r requirements.txt
 
@@ -21,15 +20,10 @@ WORKDIR $HOME/app
 
 
 ENV HF_HOME=$HOME/.cache/huggingface
-
 ENV TRANSFORMERS_CACHE=$HF_HOME
 
 
-RUN python -c "from huggingface_hub import snapshot_download; import os; snapshot_download('MBZUAI/LaMini-Flan-T5-77M', cache_dir=os.environ.get('HF_HOME'))"
-
-cache_dir = os.environ.get("HF_HOME")
-snapshot_download(repo_id="MBZUAI/LaMini-Flan-T5-77M", cache_dir=cache_dir)
-EOF
+RUN python3 -c "from huggingface_hub import snapshot_download; import os; snapshot_download('MBZUAI/LaMini-Flan-T5-77M', cache_dir=os.environ.get('HF_HOME'))"
 
 
 COPY --chown=user . $HOME/app