wolof-subtiles-generator permet de générer des sous-titres en wolof pour des fichiers audio et de créer des vidéos avec les sous-titres incrustés. L'API utilise le modèle de traitement automatique du langage naturel whisper-small-wolof pour générer les transcriptions et les sous-titres.
# Wolof Subtitles Generator API
Wolof Subtitles Generator permet de générer des sous-titres en wolof pour des fichiers audio et de créer des vidéos avec les sous-titres incrustés. L'API utilise le modèle de traitement automatique du langage naturel whisper-small-wolof pour générer les transcriptions et les sous-titres.
## Prérequis
Avant de commencer, assurez-vous d'avoir les éléments suivants :
1. **Système d'exploitation Linux**: Cette API a été conçue pour fonctionner sur un système d'exploitation Linux.
2. **ffmpeg**: Assurez-vous que le logiciel `ffmpeg` est installé sur votre système. Cet outil est nécessaire pour le traitement audio et vidéo. Si vous ne l'avez pas déjà installé, vous pouvez le faire en utilisant la commande suivante :
```bash
sudo apt-get install ffmpeg
```
3. **Dépendances Python**: Installez les dépendances Python nécessaires en exécutant la commande suivante :
```bash
python3 -m venv env
source env/bin/activate
pip install -r requirements.txt
```
4. **Dépendances Modele**: Téléchargez le modèle directement depuis google drive à l'aide de la commande gdown :
```bash
pip3 install gdown
cd whisper-wolof
gdown "
drive.google.com"
```
## Configuration
La configuration de l'API peut être ajustée en modifiant les paramètres dans le fichier `main.py`.
- `host`: L'adresse IP sur laquelle l'API écoutera les requêtes. Par défaut, elle est configurée pour écouter sur toutes les interfaces réseau (`0.0.0.0`).
- `port`: Le port sur lequel l'API écoutera les requêtes. Par défaut, le port est `8080`.
- `model`: Le chemin vers le modèle Whisper à utiliser pour la génération de sous-titres. Par défaut, le chemin est configuré pour utiliser `./whisper-wolof`.
- `device`: L'appareil à utiliser pour l'inférence du modèle. Par défaut, l'appareil est configuré pour utiliser le CPU (`cpu`).
## Utilisation
1. Lancez l'API en exécutant la commande suivante :
```bash
python main.py --host --port --mod …