Esta API permite gerar legendas para imagens a partir de URLs e, opcionalmente, traduzir as legendas para um idioma de destino especificado.
- Geração de legendas de imagens usando o modelo Blip.
- Tradução de texto gerado para diferentes idiomas.
- Endpoint
/caption/na API FastAPI. - Tratamento de erros para casos em que o idioma de destino não é suportado.
- Python
- FastAPI
- Transformers
- Googletrans
- Clone este repositório.
- Crie um ambiente virtual:
python -m venv venv - Ative o ambiente virtual:
source venv/bin/activate - Instale as dependências:
pip install -r requirements.txt - Execute a API:
fastapi dev src/main.py
Parâmetros:
image(string, obrigatório): URL da imagem.lang(string, opcional): Código do idioma para o qual a legenda deve ser traduzida.
Exemplo de requisição:
/caption/?image=https://example.com/image.jpg&lang=pt
Exemplo de resposta:
{
"image": "https://example.com/image.jpg",
"caption": "Uma descrição da imagem em português."
}A API retorna um erro 400 se o idioma especificado não for suportado.
{
"message": "Idioma não encontrado"
}