Upload 3 files

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.10
+
+WORKDIR /python-docker
+
+COPY requirements.txt requirements.txt
+RUN apt-get update && apt-get install git -y
+RUN pip3 install -r requirements.txt
+RUN apt-get update && apt-get install -y ffmpeg
+RUN python -c "from transformers import pipeline; pipeline('automatic-speech-recognition', model='openai/whisper-small')"
+COPY . .
+
+EXPOSE 8001
+
+CMD ["uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "8001"]

fastapi_app.py

@@ -0,0 +1,49 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse, RedirectResponse
+from tempfile import NamedTemporaryFile
+from typing import List
+from transformers import pipeline
+
+# Specify the model name
+model_name = "openai/whisper-small"
+
+whisper_asr = pipeline(
+        "automatic-speech-recognition",
+        model = model_name,
+        chunk_length_s = 30,
+        device = "cpu"
+    )
+
+
+app = FastAPI()
+
+@app.post("/whisper/")
+async def handler(files: List[UploadFile] = File(...)):
+    if not files:
+        raise HTTPException(status_code=400, detail="No files were provided")
+
+    # For each file, let's store the results in a list of dictionaries.
+    results = []
+
+    for file in files:
+        # Create a temporary file.
+        with NamedTemporaryFile(delete=True) as temp:
+            # Write the user's uploaded file to the temporary file.
+            with open(temp.name, "wb") as temp_file:
+                temp_file.write(file.file.read())
+            
+            # Let's get the transcript of the temporary file.
+            result = whisper_asr(temp.name, return_timestamps=False)
+
+            # Now we can store the result object for this file.
+            results.append({
+                'filename': file.filename,
+                'transcript': result['text'],
+            })
+
+    return JSONResponse(content={'results': results})
+
+
+@app.get("/", response_class=RedirectResponse)
+async def redirect_to_docs():
+    return "/docs"

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+aiofiles
+python-multipart
+uvicorn
+transformers
+torch