HARI PREETHAM commited on
Commit
4fb8564
·
1 Parent(s): 5172015

\Revert Whisper to large-v2 for verified accuracy (large-v3 hallucinated), keeping NLLB-3.3B"

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. app.py +2 -2
Dockerfile CHANGED
@@ -18,8 +18,8 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
18
  RUN mkdir -p /home/user/.cache/whisper
19
  ENV XDG_CACHE_HOME="/home/user/.cache"
20
 
21
- # Pre-download large-v3 model during build to avoid runtime timeout
22
- RUN python3 -c "from faster_whisper import download_model; download_model('large-v3')"
23
 
24
  COPY --chown=user . /app
25
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
18
  RUN mkdir -p /home/user/.cache/whisper
19
  ENV XDG_CACHE_HOME="/home/user/.cache"
20
 
21
+ # Pre-download large-v2 model (User preferred over v3 for accuracy)
22
+ RUN python3 -c "from faster_whisper import download_model; download_model('large-v2')"
23
 
24
  COPY --chown=user . /app
25
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
app.py CHANGED
@@ -29,8 +29,8 @@ app.add_middleware(
29
  allow_headers=["*"],
30
  )
31
 
32
- # Use 'large-v3' (Pre-downloaded in Dockerfile for stability & max accuracy)
33
- MODEL_SIZE = "large-v3"
34
  model = None
35
 
36
  def load_model():
 
29
  allow_headers=["*"],
30
  )
31
 
32
+ # Use 'large-v2' (User selected as most accurate, pre-downloaded in Dockerfile)
33
+ MODEL_SIZE = "large-v2"
34
  model = None
35
 
36
  def load_model():