Hugging Face's logo

Spaces:
srivatsavdamaraju
/
midassdpethflask
Runtime error

App Files Community
srivatsavdamaraju commited on
Commit
8ac5f8b
·
verified ·
1 Parent(s): a64f32c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -5,7 +5,8 @@ FROM python:3.9-slim
5
  ENV PYTHONDONTWRITEBYTECODE 1
6
  ENV PYTHONUNBUFFERED 1
7
 
8
- # Step 3: Disable torch hub cache
 
9
  ENV TORCH_HUB_NO_CACHE=1
10
 
11
  # Step 4: Install system dependencies
 
5
  ENV PYTHONDONTWRITEBYTECODE 1
6
  ENV PYTHONUNBUFFERED 1
7
 
8
+ # Step 3: Disable torch hub cache and set the cache directory
9
+ ENV TORCH_HOME=/app/.cache
10
  ENV TORCH_HUB_NO_CACHE=1
11
 
12
  # Step 4: Install system dependencies