Add application file

Dockerfile

@@ -0,0 +1,26 @@
+FROM python:3.12-slim
+
+RUN apt-get update && apt-get install -y \
+    libsndfile1-dev \
+    build-essential \
+    libopenblas-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+
+# Crear y configurar un usuario no root
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+    
+WORKDIR $HOME/app
+
+ENV HF_HOME=$HOME/.cache
+
+COPY --chown=user . $HOME/app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,6 +1,6 @@
 ---
 title: Api Dockerized Whisper
-emoji: 🔥
+emoji: 🐋
 colorFrom: blue
 colorTo: gray
 sdk: docker

app.py

@@ -0,0 +1,18 @@
+import io
+import librosa
+from transformers import pipeline
+from fastapi import FastAPI, File, UploadFile
+
+
+app = FastAPI()
+pipe = pipeline(model="pollitoconpapass/whisper-small-finetuned")
+
+@app.post("/transcribe-whisper")
+async def transcribe(audio: UploadFile = File(...)):
+    contents = await audio.read()
+    buffer = io.BytesIO(contents)
+    with buffer:
+        audio_array, _= librosa.load(buffer, sr=16000)
+
+    text = pipe(audio_array)["text"]
+    return {"text": text}

docker-compose.yml

@@ -0,0 +1,6 @@
+version: '3.12'
+services:
+  api-hf-whisper:
+    build: .
+    ports:
+      - "7860:7860"

requirements.txt

@@ -0,0 +1,7 @@
+uvicorn==0.25.0
+librosa==0.10.1
+transformers==4.39.0
+fastapi==0.108.0
+soundfile==0.12.1
+torch==2.3.0
+python-multipart