Update Dockerfile

Dockerfile

@@ -1,30 +1,35 @@
-# Start with an official PyTorch Docker image with CUDA support for GPU acceleration
+# Base image with PyTorch and CUDA for GPU support
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 
-# Install essential packages
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Hugging Face Transformers, TorchScript, and additional dependencies
-RUN pip install torch torchvision transformers requests
+# 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
 
-# Clone Meta Sapiens Pose model and MotionBERT model from Hugging Face
+# Create a working directory for model files
 RUN mkdir /models
 WORKDIR /models
 
-# Meta Sapiens Pose Model
+# Download Meta Sapiens Pose model from Hugging Face Hub
 RUN git clone https://huggingface.co/facebook/sapiens-pose-1b-torchscript /models/sapiens_pose
 
-# MotionBERT Model
+# Download MotionBERT model from Hugging Face Hub
 RUN git clone https://huggingface.co/walterzhu/MotionBERT /models/motionbert
 
-# Copy the inference script
+# Copy the inference script (app.py) into the container
 COPY app.py /app/app.py
 
-# Expose port for Flask app
+# Expose the default port for Flask
 EXPOSE 7860
 
-# Run the inference script
+# Run the Flask app
 CMD ["python", "/app/app.py"]