Skip to content

vijaygodhasara/image_caption_AI_app

Repository files navigation

Image Caption AI App

This project is an AI-powered application that generates captions for images using deep learning models.

Image screenshot

Features

  • Upload an image, and the app will generate a descriptive caption.
  • Powered by state-of-the-art machine learning libraries like transformers and torch.

Prerequisites

Make sure you have the following installed:

  • Python 3.7+
  • Required libraries (listed in requirements.txt)

Technologies Used

  • Python
  • Gradio: For building the web interface
  • Transformers: Pre-trained models from Hugging Face for image captioning
  • Torch: Deep learning framework

Installation

  1. Clone the repository:

    git clone https://github.com/vijaygodhasara/image_caption_AI_app.git
    
  2. Navigate into the project directory:

    cd image_caption_AI_app
    
  3. Install the required dependencies:

    pip install -r requirements.txt
    

Usage

Once all dependencies are installed, you can run the application with the following command:

python image_captioning_app.py

About

image captioning generative AI app

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages