Upload folder using huggingface_hub

Dockerfile

@@ -1,45 +1,73 @@
-# NVIDIA CUDA base image for A10G Large GPU
+# Production Docker image with all AI models and dependencies
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-# Set environment variables
+# Prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
     GRADIO_SERVER_NAME=0.0.0.0 \
     GRADIO_SERVER_PORT=7860 \
-    CUDA_HOME=/usr/local/cuda
+    CUDA_HOME=/usr/local/cuda \
+    HF_HOME=/home/user/.cache/huggingface
 
-# Install Python 3.10 and essential tools
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     python3.10 \
     python3-pip \
+    python3-dev \
+    build-essential \
+    git \
+    curl \
+    wget \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
     && rm -rf /var/lib/apt/lists/* \
     && ln -s /usr/bin/python3.10 /usr/bin/python
 
-# Set up user with ID 1000 (required by HuggingFace Spaces)
+# Create user with ID 1000
 RUN useradd -m -u 1000 user
 
 # Switch to user
 USER user
 
-# Set home and path
+# Set paths
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
 # Set working directory
 WORKDIR $HOME/app
 
-# Copy requirements
+# Copy requirements first (for Docker layer caching)
 COPY --chown=user requirements.txt .
 
-# Install Python packages
-RUN pip install --no-cache-dir --upgrade pip && \
+# Install all Python dependencies
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir gradio && \
+    pip install --no-cache-dir transformers accelerate diffusers && \
+    pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
+    pip install --no-cache-dir opencv-python-headless pillow numpy scipy && \
+    pip install --no-cache-dir huggingface_hub hf-transfer && \
+    pip install --no-cache-dir sentencepiece protobuf && \
     pip install --no-cache-dir -r requirements.txt
 
-# Copy application
+# Copy the complete application
 COPY --chown=user app.py .
 
-# Expose Gradio port
+# Create necessary directories
+RUN mkdir -p $HOME/app/data $HOME/.cache/huggingface
+
+# Pre-download critical models (optional - uncomment if needed)
+# RUN python -c "from huggingface_hub import snapshot_download; \
+#     snapshot_download('Qwen/Qwen2.5-7B-Instruct', cache_dir='$HF_HOME')"
+
+# Expose port
 EXPOSE 7860
 
-# Run the Gradio application
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=120s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
+# Launch Gradio app
 CMD ["python", "app.py"]

requirements.txt

@@ -1 +1,28 @@
+# Core Gradio
 gradio>=4.0
+
+# AI/ML Frameworks
+transformers>=4.30.0
+accelerate
+diffusers
+sentencepiece
+protobuf
+
+# Image Processing
+opencv-python-headless
+pillow
+imageio
+
+# Audio Processing
+librosa
+soundfile
+
+# Data Science
+numpy
+pandas
+scipy
+
+# Utilities
+huggingface_hub
+requests
+aiohttp