Update Dockerfile

Dockerfile

@@ -1,16 +1,36 @@
-# Use the edgeneko/neko-image-gallery image as base
-FROM edgeneko/neko-image-gallery:latest-cpu
+ARG TORCH_VERSION=2.3.0
+ARG CUDA_VERSION=12.1
+FROM pytorch/pytorch:${TORCH_VERSION}-cuda${CUDA_VERSION}-cudnn8-runtime
 
-# Set working directory
 WORKDIR /opt/NekoImageGallery
 
-VOLUME ["/opt/NekoImageGallery/static"]
+COPY requirements.txt .
+
+RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -r requirements.txt
+
+RUN mkdir -p /opt/models && \
+    export PYTHONDONTWRITEBYTECODE=1 && \
+    huggingface-cli download openai/clip-vit-large-patch14 'model.safetensors' '*.txt' '*.json' --local-dir /opt/models/clip && \
+    huggingface-cli download google-bert/bert-base-chinese 'model.safetensors' '*.txt' '*.json' --local-dir /opt/models/bert && \
+    huggingface-cli download pk5ls20/PaddleModel 'PaddleOCR2Pytorch/ch_ptocr_v4_det_infer.pth' 'PaddleOCR2Pytorch/ch_ptocr_v4_rec_infer.pth' \
+     'PaddleOCR2Pytorch/ch_ptocr_mobile_v2.0_cls_infer.pth' 'PaddleOCR2Pytorch/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml' \
+     'PaddleOCR2Pytorch/configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml' 'ppocr_keys_v1.txt' --local-dir /opt/models/ocr && \
+    rm -rf /root/.cache/huggingface
+
+ENV APP_MODEL__CLIP=/opt/models/clip
+ENV APP_MODEL__BERT=/opt/models/bert
+ENV APP_MODEL__EASYPADDLEOCR=/opt/models/ocr
 
-# Copy static files
-# COPY ./static /opt/NekoImageGallery/static
+COPY . .
+
+EXPOSE 8000
+
+VOLUME ["/opt/NekoImageGallery/static"]
 
-# Expose port
-EXPOSE 3000
+LABEL org.opencontainers.image.authors="EdgeNeko" \
+      org.opencontainers.image.url="https://github.com/hv0905/NekoImageGallery" \
+      org.opencontainers.image.source="https://github.com/hv0905/NekoImageGallery" \
+      org.opencontainers.image.title="NekoImageGallery" \
+      org.opencontainers.image.description="An AI-powered natural language & reverse Image Search Engine powered by CLIP & qdrant."
 
-# Start the application
-CMD ["python", "app.py"]
+ENTRYPOINT ["python", "main.py"]