diff --git a/Dockerfile b/Dockerfile
index 2d38793..e24af03 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,4 +2,6 @@ FROM ubuntu:latest
 
 RUN apt update && apt install -y python3-pip && pip3 install scikit-learn datasets
 
+ENV HF_HOME="/tmp/hf"
+
 COPY . .
\ No newline at end of file