mirror of
https://github.com/azaion/detections.git
synced 2026-04-22 11:26:33 +00:00
Oleksandr Bezdieniezhnykh
8ce40a9385
- Introduced `AIAvailabilityStatus` class to manage the availability status of AI models, including methods for setting status and logging messages. - Added `AIRecognitionConfig` class to encapsulate configuration parameters for AI recognition, with a static method for creating instances from dictionaries. - Implemented enums for AI availability states to enhance clarity and maintainability. - Updated related Cython files to support the new classes and ensure proper type handling. These changes aim to improve the structure and functionality of the AI model management system, facilitating better status tracking and configuration handling.
11 lines
457 B
Docker
11 lines
457 B
Docker
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
|
|
RUN apt-get update && apt-get install -y python3 python3-pip python3-dev gcc libgl1 libglib2.0-0 && rm -rf /var/lib/apt/lists/*
|
|
WORKDIR /app
|
|
COPY requirements.txt requirements-gpu.txt ./
|
|
RUN pip3 install --no-cache-dir -r requirements-gpu.txt
|
|
COPY . .
|
|
RUN python3 setup.py build_ext --inplace
|
|
ENV PYTHONPATH=/app/src
|
|
EXPOSE 8080
|
|
CMD ["python3", "-m", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8080"]
|