adam/cartman
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

docker

Browse source
This commit is contained in:
Adam 2023-02-09 18:05:05 -05:00
parent 00c6d77380
commit 9d303d99d3
3 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
api/Dockerfile Normal file
View file

@ -0,0 +1,7 @@
# syntax-docker/dockerfile:1
FROM pytorch/pytorch
WORKDIR /app
COPY requirements.txt requirements.txt
RUN pip3 install -r requirements.txt
COPY . .
CMD [ "uvicorn", "chatbots:api", "--proxy-headers", "--host", "0.0.0.0", "--port", "8000"]

9
api/requirements.txt
View file

@ -1,6 +1,3 @@
fastapi>=0.89.1 fastapi
torch>=1.13.1 uvicorn
torchaudio>=0.13.1 transformers
torchvision>=0.14.1
transformers>=4.26.0
uvicorn>=0.20.0

2
api/src/bots/cartman.py
View file

@ -7,7 +7,7 @@ tokenizer = AutoTokenizer.from_pretrained(
"microsoft/DialoGPT-large", padding_side='left' "microsoft/DialoGPT-large", padding_side='left'
) )
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
"../train/cartman/models/output-medium" "src/bots/cartman"
) )