Update Dockerfile

Dockerfile

@@ -5,8 +5,8 @@ FROM python:3.9-slim
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
-# Step 3: Set the environment variable for PyTorch cache
-ENV TORCH_HOME=/app/.cache
+# Step 3: Disable torch hub cache
+ENV TORCH_HUB_NO_CACHE=1
 
 # Step 4: Install system dependencies
 RUN apt-get update && apt-get install -y \