Prashant26am commited on
Commit
6fc2918
·
1 Parent(s): 7fceff6

Fix: Update Whisper package installation and imports

Browse files
Files changed (3) hide show
  1. Dockerfile +2 -0
  2. app.py +1 -1
  3. requirements.txt +4 -2
Dockerfile CHANGED
@@ -5,6 +5,8 @@ WORKDIR /code
5
  # Install system dependencies
6
  RUN apt-get update && apt-get install -y \
7
  ffmpeg \
 
 
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
  COPY requirements.txt .
 
5
  # Install system dependencies
6
  RUN apt-get update && apt-get install -y \
7
  ffmpeg \
8
+ git \
9
+ build-essential \
10
  && rm -rf /var/lib/apt/lists/*
11
 
12
  COPY requirements.txt .
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from fastapi import FastAPI, UploadFile, File, Form, HTTPException
2
  from fastapi.responses import JSONResponse
3
- import openai.whisper as whisper
4
  import tempfile
5
  import os
6
  from transformers import pipeline
 
1
  from fastapi import FastAPI, UploadFile, File, Form, HTTPException
2
  from fastapi.responses import JSONResponse
3
+ import whisper
4
  import tempfile
5
  import os
6
  from transformers import pipeline
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
  fastapi
2
  uvicorn
3
- openai-whisper
4
  torch
5
  transformers
6
  pydantic
7
- python-multipart
 
 
 
1
  fastapi
2
  uvicorn
3
+ git+https://github.com/openai/whisper.git
4
  torch
5
  transformers
6
  pydantic
7
+ python-multipart
8
+ numpy
9
+ tqdm