mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-23 10:33:51 +00:00
26 lines
525 B
Docker
26 lines
525 B
Docker
FROM nvcr.io/nvidia/pyg:25.03-py3
|
|
|
|
WORKDIR /app
|
|
|
|
# Install Flask and other required packages
|
|
RUN pip install --no-cache-dir \
|
|
flask==2.0.1 \
|
|
gunicorn==23.0.0 \
|
|
tqdm
|
|
|
|
# Create model directory
|
|
RUN mkdir -p /app/models
|
|
|
|
# Copy application code
|
|
COPY services/gnn_model/app.py /app/
|
|
|
|
# Set environment variables
|
|
ENV MODEL_PATH=/app/models/tech-qa-model.pt
|
|
ENV PYTHONUNBUFFERED=1
|
|
ENV FLASK_APP=app.py
|
|
|
|
# Expose the port
|
|
EXPOSE 5000
|
|
|
|
# Run the service with gunicorn
|
|
CMD ["gunicorn", "--bind", "0.0.0.0:5000", "app:app"] |