Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import os | |
| def generer_legende_en_francais(chemin_image): | |
| """ | |
| Génère une légende pour une image donnée en anglais et la traduit en français. | |
| :param chemin_image: Le chemin vers l'image pour laquelle générer une légende. | |
| :type chemin_image: str | |
| :return: La légende traduite en français. | |
| :rtype: str | |
| """ | |
| # Initialiser la pipeline de légendage d'image | |
| pipeline_legende_image = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large") | |
| # Initialiser la pipeline de traduction de l'anglais vers le français | |
| pipeline_traduction = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr") | |
| # Génération de la légende en anglais | |
| resultats_legende = pipeline_legende_image(chemin_image) | |
| legende_anglaise = resultats_legende[0]['generated_text'] | |
| # Traduction de la légende en français | |
| legende_francaise = pipeline_traduction(legende_anglaise, max_length=512) | |
| texte_francais = legende_francaise[0]['translation_text'] | |
| return texte_francais | |
| #generer_legende_en_francais('/content/IMG-20221113-WA0034.jpg') | |
| gr.Interface(fn=generer_legende_en_francais, | |
| inputs=gr.Image(type="pil"), | |
| outputs="text").launch() | |