cog-batch-image-captioning

A cog model for batch image captioning using various AI from OpenAI, Anthropic, and Google's Generative AI:

Features

Process multiple images from a ZIP archive
supports png, jpg, jpeg, webp
Optional image resizing for more cost-effective captioning
Customizable caption prefixes and suffixes
Support for multiple AI models:
- OpenAI: GPT-4 and variants
- Anthropic: Claude-3.5, Claude-3 variants
- Google: Gemini-1.5 variants
Flexible system and message prompts
Error handling and retry mechanism
Output as a ZIP file containing captions that match image filenames as well as a CSV summary

Name		Name	Last commit message	Last commit date
Latest commit History 20 Commits
.dockerignore		.dockerignore
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
captioner.py		captioner.py
cog.yaml		cog.yaml
operation.py		operation.py
predict.py		predict.py