Ce script Python lit les légendes d'image à partir d'un fichier CSV, les raccourcit à l'aide de l'API d'Openai, puis les enregistre dans un fichier JSON. Cela pourrait être particulièrement utile si vous avez besoin de légendes d'image égales ou inférieures à un certain nombre de caractères pour l'accessibilité, les médias sociaux ou la formation d'un modèle d'image.
Charge en toute sécurité les clés API et autres paramètres sensibles à partir d'un fichier .env.
Utilise le modèle GPT-3.5-turbo d'OpenAI pour raccourcir efficacement les légendes.
Lit des légendes à partir d'un fichier CSV, qui permet le traitement par lots de plusieurs légendes.
Sorte les légendes traitées dans un format JSON, ce qui facilite l'intégration avec des applications Web ou d'autres supports. Configuration et utilisation
python -m venv venv
source venv/bin/activate # On Windows use `venvScriptsactivate`
pip install openai python-dotenv
Créez un fichier .env dans le même répertoire que le script et ajoutez votre touche API OpenAI:
OPENAI_API_KEY='your_api_key_here'
Assurez-vous que votre fichier CSV est formaté correctement avec le nom de fichier et les en-têtes de légende.
Modifiez le script pour inclure le chemin correct vers votre fichier CSV dans l'appel process_captions («votre chemin ici»).
La sortie sera dans captions.json .