Animals Classifier

This is a simple image classifier that receives an image and if it is an animal, provides the name, description, and determines if the animal is dangerous based on Wikipedia info.

Getting Started

Prerequisites

You need to have the following installed on your machine:

• Python 3.12 or later
• Pip
• Virtualenv

How to use it

Running the script

# Clone the repository
git clone

# Change directory
cd animals-classifier

# Install virtualenv (in case you don't have it)
pip install virtualenv

# Create a virtual environment
python -m venv my_virtual_environment

# Activate the virtual environment
source my_virtual_environment/bin/activate

# Install the dependencies
pip install -r requirements.txt

# Run the main script
python main.py

Open the browser and go to the address shown in the terminal: http://127.0.0.1:7860

Important: Make sure to set your OpenAI key before running: export OPENAI_API_KEY-="sk-..."

Resources

• Images:
  • See some images for testing in the images folder.
  • Taken from Unsplash, under Free License.

Built with

• Python
• Gradio
• Pillow
• OpenAI
• LlamaIndex

Related Docs

• VectorStoreIndex
• OpenAI Embeddings
• WikipediaReader