Upload Dockerfile

Dockerfile

@@ -13,11 +13,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the rest of the application into the container at /app
 COPY . /app
 
-# Create a cache directory for transformers
-RUN mkdir -p /app/cache
+# Create a cache directory for Hugging Face
+RUN mkdir -p /app/hf_cache
 
-# Set environment variable for transformers cache
-ENV TRANSFORMERS_CACHE=/app/cache
+# Set environment variable for Hugging Face cache
+ENV HF_HOME=/app/hf_cache
 
 # Expose port 7860
 EXPOSE 7860