Spaces:
Running
Running
| 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 | |