From 74c9d4ea440807be9ebc2d4ee1dbb64f986c7c82 Mon Sep 17 00:00:00 2001 From: julian Date: Mon, 27 Jan 2025 15:50:43 -0500 Subject: [PATCH] adds Dockerfile --- .gitignore | 3 ++- Dockerfile | 16 ++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 Dockerfile diff --git a/.gitignore b/.gitignore index 229e98c..656557b 100644 --- a/.gitignore +++ b/.gitignore @@ -418,4 +418,5 @@ tags [._]*.un~ .vscode .github -generated_samples/ \ No newline at end of file +generated_samples/ +huggingface diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..89dbf52 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,16 @@ +# 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"] +