Spaces:
Sleeping
Sleeping
File size: 1,413 Bytes
47c1b66 d35bcf6 47c1b66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# Use a lightweight Python base image
FROM python:3.9-slim
# Set working directory
WORKDIR /app
# Set environment variables
ENV HF_HOME="/tmp/huggingface/cache"
ENV FLASK_APP=app.py
ENV FLASK_ENV=production
ENV PYTHONUNBUFFERED=1
ENV CUDA_VISIBLE_DEVICES="-1"
# Create Hugging Face cache directory with proper permissions
RUN mkdir -p /tmp/huggingface/cache && \
chmod -R 777 /tmp/huggingface/cache && \
apt-get update && \
apt-get install -y --no-install-recommends gcc python3-dev && \
rm -rf /var/lib/apt/lists/*
# Copy requirements and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Copy application code
COPY . .
# Create models directory if it doesn't exist
RUN mkdir -p models
# Ensure permissions for application files
RUN chmod -R 755 /app
# Expose application port (Hugging Face Spaces uses port 7860)
EXPOSE 7860
# Create a script to run the application
RUN echo '#!/bin/bash\n\
# Check if model files exist\n\
if [ ! -f "models/mobilenetv2_ham10000_finetuned_74.h5" ] || [ ! -f "models/efficientnetb0_skin_cancer_model_final_69.keras" ]; then\n\
echo "Warning: Model files not found. Please ensure models are uploaded to the models directory."\n\
fi\n\
\n\
# Run the Flask application\n\
flask run --host=0.0.0.0 --port=7860\n\
' > /app/start.sh && chmod +x /app/start.sh
# Run the Flask application
CMD ["/app/start.sh"]
|