Spaces:
Sleeping
Sleeping
# Base image with PyTorch and CUDA for GPU support | |
FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime | |
# Install system dependencies | |
RUN apt-get update && apt-get install -y \ | |
git \ | |
wget \ | |
&& rm -rf /var/lib/apt/lists/* | |
# Install Python packages including Hugging Face Transformers, TorchScript, and Flask | |
RUN pip install --no-cache-dir \ | |
torch \ | |
torchvision \ | |
transformers \ | |
requests \ | |
Flask # Explicitly add Flask here | |
# Create a working directory for model files | |
RUN mkdir /models | |
WORKDIR /models | |
# Download Meta Sapiens Pose model from Hugging Face Hub | |
RUN git clone https://huggingface.co/facebook/sapiens-pose-1b-torchscript /models/sapiens_pose | |
# Download MotionBERT model from Hugging Face Hub | |
RUN git clone https://huggingface.co/walterzhu/MotionBERT /models/motionbert | |
# Copy the inference script (app.py) into the container | |
COPY app.py /app/app.py | |
# Expose the default port for Flask | |
EXPOSE 7860 | |
# Run the Flask app | |
CMD ["python", "/app/app.py"] | |