1.0.1

Dockerfile

@@ -9,28 +9,19 @@ RUN apt-get update && apt-get install -y \
     git \
     wget \
     curl \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    libgomp1 \
-    libglib2.0-0 \
-    libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
-# Install Python dependencies
+# Install Python dependencies (CPU-only PyTorch)
 RUN pip install --no-cache-dir --upgrade pip && \
+    pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu && \
     pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY . .
 
-# Create necessary directories
-RUN mkdir -p /app/checkpoints /app/temp
-
 # Set environment variables
 ENV PYTHONPATH=/app
 ENV TORCH_HOME=/tmp/torch

requirements.txt

@@ -1,15 +1,10 @@
-fastapi==0.104.1
-uvicorn[standard]==0.24.0
-transformers==4.36.0
-torch==2.1.0+cpu
-torchvision==0.16.0+cpu
-opencv-python-headless==4.8.1.78
-Pillow==10.1.0
-numpy==1.24.3
-huggingface-hub==0.19.4
-timm==0.9.12
-matplotlib==3.8.2
-requests==2.31.0
-python-multipart==0.0.6
-accelerate
-python-multipart
+fastapi
+uvicorn[standard]
+transformers
+opencv-python-headless
+Pillow
+numpy
+huggingface-hub
+requests
+python-multipart
+accelerate