This project is an AI-powered application that generates captions for images using deep learning models.
- Upload an image, and the app will generate a descriptive caption.
- Powered by state-of-the-art machine learning libraries like
transformers
andtorch
.
Make sure you have the following installed:
- Python 3.7+
- Required libraries (listed in
requirements.txt
)
- Python
- Gradio: For building the web interface
- Transformers: Pre-trained models from Hugging Face for image captioning
- Torch: Deep learning framework
-
Clone the repository:
git clone https://github.com/vijaygodhasara/image_caption_AI_app.git
-
Navigate into the project directory:
cd image_caption_AI_app
-
Install the required dependencies:
pip install -r requirements.txt
Once all dependencies are installed, you can run the application with the following command:
python image_captioning_app.py