Skip to content

alinrajpoot/captnr

Repository files navigation

Captnr

Open-source tool for generating descriptive image captions and alt-text using AI. Built with Hugging Face transformers and Gradio.

Features

  • 🖼️ Image Caption Generation: Generate descriptive captions for images using AI
  • 🚀 Gradio Interface: Fast, modern AI-Powered apps interface with the API ready feature
  • 🤗 Hugging Face Integration: Uses state-of-the-art BLIP model for image captioning
  • 📦 Batch Processing: Support for processing multiple images at once
  • 🔥 GPU Support: Automatic GPU acceleration when available

Quick Start

Installation

  1. Clone the repository:
git clone https://github.com/alinrajpoot/captnr.git
cd captnr
  1. Install dependencies:
pip install -r requirements.txt
  1. Run the server:
python main.py

The API will be available at http://localhost:9000

Usage

Once the server is running, visit http://localhost:9000 in browser.

Requirements

  • Python 3.8+
  • Gradio
  • Hugging Face Transformers
  • PyTorch
  • Pillow (PIL)

Model Information

This tool uses the Salesforce/blip-image-captioning-base model from Hugging Face, which provides high-quality image captions suitable for accessibility and content description purposes.

License

Open source - feel free to contribute and improve!

About

Open-source tool and library for generating descriptive image captions and alt-text using AI.

Topics

Resources

License

Stars

Watchers

Forks