mirror of
https://github.com/deepseek-ai/Janus.git
synced 2025-02-23 14:18:58 -05:00
17 lines
399 B
Docker
17 lines
399 B
Docker
|
# Use the PyTorch base image
|
||
|
|
FROM pytorch/pytorch:latest
|
||
|
|
|
||
|
|
# Set the working directory inside the container
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
# Copy the current directory into the container
|
||
|
|
COPY . /app
|
||
|
|
|
||
|
|
# Install necessary Python packages
|
||
|
|
RUN pip install -e . && \
|
||
|
|
pip install fastapi python-multipart uvicorn
|
||
|
|
|
||
|
|
# Set the entrypoint for the container to launch your FastAPI app
|
||
|
|
CMD ["python", "demo/fastapi_app.py"]
|
||
|
|
|