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"] +