copy downloaded model
This commit is contained in:
@@ -1,16 +1,18 @@
|
||||
FROM python:3.10-slim
|
||||
|
||||
# Install dependencies
|
||||
RUN apt-get update && apt-get install -y \
|
||||
build-essential \
|
||||
ffmpeg \
|
||||
curl \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy app and model
|
||||
COPY app ./app
|
||||
|
||||
EXPOSE 4002
|
||||
|
||||
@@ -5,8 +5,8 @@ from whispercpp import Whisper
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
# Load the model (auto-download and cache it)
|
||||
whisper = Whisper.from_pretrained("base.en") # Options: tiny, base, small, etc.
|
||||
# Load the local model without Hugging Face token
|
||||
whisper = Whisper.from_pretrained("./app/model/ggml-base.en.bin")
|
||||
|
||||
@app.post("/transcribe")
|
||||
async def transcribe_audio(audio: UploadFile = File(...)):
|
||||
|
||||
Reference in New Issue
Block a user