Spaces:
Sleeping
Sleeping
fix 2
Browse files- Dockerfile +4 -3
- 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
|