Upload 4 files

.huggingface/README.md

@@ -0,0 +1 @@
+This is a Bark API for text-to-speech conversion.

Dockerfile

@@ -0,0 +1,10 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI, Body
+from huggingface_hub import snapshot_download
+from transformers import pipeline
+
+app = FastAPI()
+
+# Télécharge dynamiquement ton modèle Bark depuis ton Model Repo
+model_dir = snapshot_download("suno/bark-small", repo_type="model")
+
+tts = pipeline("text-to-speech", model=model_dir)
+
+@app.post("/tts")
+async def tts_api(text: str = Body(..., embed=True)):
+    out = tts(text, forward_params={"do_sample": True})
+    return {"sampling_rate": out["sampling_rate"], "audio": out["audio"].tolist()}

requirements.txt

@@ -0,0 +1,5 @@
+fastapi==0.99.1
+uvicorn[standard]
+transformers
+huggingface_hub
+torch