Update Dockerfile

Dockerfile

@@ -2,11 +2,8 @@ FROM python:3.9
 
 WORKDIR /code
 
-# Create the cache directory
-RUN mkdir -p /code/hf_cache
-
-# Set the HF_HOME environment variable to the cache directory
-ENV HF_HOME=/code/hf_cache
+# Set the HF_HOME environment variable to the working directory
+ENV HF_HOME=/code
 
 COPY ./requirements.txt /code/requirements.txt