hbs2 commited on
Commit
a1780ec
1 Parent(s): ff3b378

Upload 11 files

Browse files
Files changed (3) hide show
  1. Dockerfile +4 -2
  2. launcher.py +7 -1
  3. subgen.env +2 -2
Dockerfile CHANGED
@@ -1,15 +1,17 @@
1
  FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
2
 
3
- WORKDIR /home/user/app
4
 
5
  ADD https://raw.githubusercontent.com/McCloudS/subgen/main/requirements.txt /subgen/requirements.txt
6
 
7
  RUN apt-get update \
8
  && apt-get install -y \
 
 
9
  ffmpeg \
10
  && apt-get clean \
11
  && rm -rf /var/lib/apt/lists/* \
12
- && -r requirements.txt
13
 
14
  ENV PYTHONUNBUFFERED=1
15
 
 
1
  FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
2
 
3
+ WORKDIR /subgen
4
 
5
  ADD https://raw.githubusercontent.com/McCloudS/subgen/main/requirements.txt /subgen/requirements.txt
6
 
7
  RUN apt-get update \
8
  && apt-get install -y \
9
+ python3 \
10
+ python3-pip \
11
  ffmpeg \
12
  && apt-get clean \
13
  && rm -rf /var/lib/apt/lists/* \
14
+ && pip3 install -r requirements.txt
15
 
16
  ENV PYTHONUNBUFFERED=1
17
 
launcher.py CHANGED
@@ -14,7 +14,13 @@ def install_packages_from_requirements(requirements_file):
14
  subprocess.run(['pip3', 'install', '-r', requirements_file, '--upgrade'], check=True)
15
  print("Packages installed successfully using pip3.")
16
  except subprocess.CalledProcessError:
17
-
 
 
 
 
 
 
18
  def download_from_github(url, output_file):
19
  try:
20
  with urllib.request.urlopen(url) as response, open(output_file, 'wb') as out_file:
 
14
  subprocess.run(['pip3', 'install', '-r', requirements_file, '--upgrade'], check=True)
15
  print("Packages installed successfully using pip3.")
16
  except subprocess.CalledProcessError:
17
+ try:
18
+ # If pip3 fails, try installing with pip
19
+ subprocess.run(['pip', 'install', '-r', requirements_file, '--upgrade'], check=True)
20
+ print("Packages installed successfully using pip.")
21
+ except subprocess.CalledProcessError:
22
+ print("Failed to install packages using both pip3 and pip.")
23
+
24
  def download_from_github(url, output_file):
25
  try:
26
  with urllib.request.urlopen(url) as response, open(output_file, 'wb') as out_file:
subgen.env CHANGED
@@ -1,6 +1,6 @@
1
- WHISPER_MODEL=large-v3
2
  WEBHOOKPORT=9000
3
  TRANSCRIBE_DEVICE=gpu
4
  DEBUG=True
5
  CLEAR_VRAM_ON_COMPLETE=False
6
- APPEND=True
 
1
+ WHISPER_MODEL=medium
2
  WEBHOOKPORT=9000
3
  TRANSCRIBE_DEVICE=gpu
4
  DEBUG=True
5
  CLEAR_VRAM_ON_COMPLETE=False
6
+ APPEND=False