Upload Dockerfile with huggingface_hub

Dockerfile

@@ -2,33 +2,39 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies
+# Install dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
-    cmake \
-    git \
-    libcairo2-dev \
-    pkg-config \
     python3-dev \
     libfreetype6-dev \
+    pkg-config \
+    libpng-dev \
+    cmake \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first to leverage Docker cache
-COPY requirements.txt .
+# Clone diffvg repository
+RUN git clone https://github.com/BachiLi/diffvg.git && \
+    cd diffvg && \
+    git submodule update --init --recursive && \
+    python setup.py install
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Install PyTorch and other dependencies
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+RUN pip install transformers pillow svgwrite svgpathtools numpy tqdm scikit-image matplotlib
+RUN pip install fastapi uvicorn
 
-# Install diffvg from the DiffSketcher project
-RUN pip install --no-cache-dir git+https://github.com/ximinng/DiffSketcher-project.git#subdirectory=diffvg
+# Copy model files
+COPY . /app/model
 
-# Copy the model files
-COPY . .
+# Copy handler files
+COPY handler_template.py /app/
+COPY diffsketcher_handler.py /app/
+COPY api.py /app/
 
 # Set environment variables
-ENV PYTHONUNBUFFERED=1
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONPATH=/app
+ENV MODEL_ID="jree423/diffsketcher"
+ENV TASK="text-to-image"
 
-# Run the API server
-CMD ["python", "app.py"]
+# Run the API
+CMD ["python", "api.py"]