diff --git a/subgen/Dockerfile b/subgen/Dockerfile index 79a6231..0a02eda 100644 --- a/subgen/Dockerfile +++ b/subgen/Dockerfile @@ -1,23 +1,5 @@ FROM python -ARG WHISPER_MODEL medium -# this can be tiny, base, small, medium, large -ARG WHISPER_SPEEDUP False -# this adds the option -su "speed up audio by x2 (reduced accuracy)" -ARG WHISPER_THREADS 4 -# number of threads to use during computation -ARG WHISPER_PROCESSORS 1 -# number of processors to use during computation - -ARG PROCADDEDMEDIA True -# will gen subtitles for all media added regardless of existing external/embedded subtitles -ARG PROCMEDIAONPLAY True -# will gen subtitles for all played media regardless of existing external/embedded subtitles -ARG NAMESUBLANG aa -# use 2 letter codes @ https://en.wikipedia.org/wiki/List_of_ISO_639-1_codes // makes it easier to see which subtitles are generated by subgen -- names file like: "Community - S03E18 - Course Listing Unavailable.aa.subgen.srt" -ARG UPDATEREPO True -# pulls and merges whisper.cpp on every start - RUN apt-get update && apt-get install -y ffmpeg git gcc RUN pip3 install webhook_listener @@ -25,4 +7,4 @@ EXPOSE 8090 COPY ./subgen/subgen.py / -CMD [ "python3", "-u", "/subgen.py", "$WHISPER_MODEL", "$WHISPER_SPEEDUP", "$WHISPER_THREADS", "$WHISPER_PROCESSORS", "$PROCADDEDMEDIA", "$PROCMEDIAONPLAY", "$NAMESUBLANG", "$UPDATEREPO"] +CMD [ "python3", "-u", "/subgen.py"]