diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..baed5f4 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,2 @@ +FROM semitechnologies/transformers-inference:custom +RUN MODEL_NAME=distilroberta-base ./download.py \ No newline at end of file