README.md

metadata

title: Image Caption Generator
emoji: 🖼️
colorFrom: blue
colorTo: purple
sdk: docker
sdk_version: latest
app_file: app.py
pinned: false
license: mit

Image Caption Generator

Generate captions for images using an optimized EfficientNet-B3 model.

Features

• ✅ EfficientNet-B3 model for high-quality captions
• ✅ Optimized quantized model (~245MB)
• ✅ Fast inference
• ✅ Simple web interface

How to Use

1. Upload an image (PNG, JPG, JPEG)
2. Click "Generate Caption"
3. Get your caption!

Model

• Architecture: EfficientNet-B3
• Optimization: INT8 Quantization
• Size: ~245MB

Technical Details

• Built with PyTorch and Transformers
• Uses GPT-2 tokenizer
• Optimized for production deployment