Spaces:
Running
Running
File size: 746 Bytes
400b4a4 32d4a86 400b4a4 32d4a86 400b4a4 32d4a86 400b4a4 32d4a86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
---
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
|