aryan083 commited on
Commit
e69ac01
·
1 Parent(s): 19cd08b
Files changed (2) hide show
  1. Dockerfile +4 -3
  2. app.py +2 -0
Dockerfile CHANGED
@@ -30,9 +30,9 @@ RUN pip install --no-cache-dir -r requirements.txt
30
  # Copy application code
31
  COPY . .
32
 
33
- # Create necessary directories with proper permissions
34
  RUN mkdir -p /app/models /tmp/logs /tmp/temp && \
35
- chmod 777 /app/models /tmp/logs /tmp/temp
36
 
37
  # Set environment variables for Spaces
38
  ENV APP_ENV=prod \
@@ -41,7 +41,8 @@ ENV APP_ENV=prod \
41
  MODEL_CACHE_DIR=/app/models \
42
  HF_HOME=/app/models \
43
  TRANSFORMERS_CACHE=/app/models \
44
- TORCH_HOME=/app/models
 
45
 
46
  # Expose port for Spaces
47
  EXPOSE 7860
 
30
  # Copy application code
31
  COPY . .
32
 
33
+ # Create necessary directories with proper permissions (recursive)
34
  RUN mkdir -p /app/models /tmp/logs /tmp/temp && \
35
+ chmod -R 777 /app/models /tmp/logs /tmp/temp
36
 
37
  # Set environment variables for Spaces
38
  ENV APP_ENV=prod \
 
41
  MODEL_CACHE_DIR=/app/models \
42
  HF_HOME=/app/models \
43
  TRANSFORMERS_CACHE=/app/models \
44
+ TORCH_HOME=/app/models \
45
+ XDG_CACHE_HOME=/app/models
46
 
47
  # Expose port for Spaces
48
  EXPOSE 7860
app.py CHANGED
@@ -21,6 +21,8 @@ os.environ['GRADIO_SERVER_PORT'] = '7860'
21
  os.environ['MODEL_CACHE_DIR'] = '/app/models'
22
  os.environ['HF_HOME'] = '/app/models'
23
  os.environ['TRANSFORMERS_CACHE'] = '/app/models'
 
 
24
  os.environ['HF_TOKEN'] = os.getenv('HF_TOKEN') or os.getenv('HUGGINGFACE_HUB_TOKEN') or ""
25
 
26
  # Add src to Python path
 
21
  os.environ['MODEL_CACHE_DIR'] = '/app/models'
22
  os.environ['HF_HOME'] = '/app/models'
23
  os.environ['TRANSFORMERS_CACHE'] = '/app/models'
24
+ os.environ['TORCH_HOME'] = '/app/models'
25
+ os.environ['XDG_CACHE_HOME'] = '/app/models'
26
  os.environ['HF_TOKEN'] = os.getenv('HF_TOKEN') or os.getenv('HUGGINGFACE_HUB_TOKEN') or ""
27
 
28
  # Add src to Python path