---
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