File size: 697 Bytes
ee89c03
 
 
 
 
 
 
 
 
 
3462329
ee89c03
 
 
3462329
 
 
 
 
ee89c03
 
3462329
ee89c03
 
 
3462329
ee89c03
 
 
3462329
ee89c03
 
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
---
title: Image Captioning Demo
emoji: 📷
colorFrom: blue
colorTo: green
sdk: gradio
sdk_version: 4.44.0
app_file: app.py
pinned: false
---

# Image Captioning Demo

This Space demonstrates an Image Captioning model trained on a custom dataset with 6000 training images. Upload an image to generate a caption.

## Model Details
- **Image Encoder**: ResNet50
- **Text Decoder**: LSTM with fine-tuned GloVe embeddings
- **Vocabulary Size**: 3478
- **Max Caption Length**: 20
- **Sampling**: Top-k (k=5)

## Example
- **Input**: Image of a dog
- **Output**: "A dog is running in the park"

## Usage
1. Upload an image in PNG or JPEG format.
2. Click "Submit" to generate a caption.

## License
MIT