mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-27 04:13:52 +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==20.1.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"]
|