An open API service indexing awesome lists of open source software.

https://github.com/rixiiz/image-classifier

An AI model that can classifier uploaded images
https://github.com/rixiiz/image-classifier

artificial-intelligence cv2 image-recognition mobilenetv2 numpy openai openai-api pil streamlit tensorflow

Last synced: 3 months ago
JSON representation

An AI model that can classifier uploaded images

Awesome Lists containing this project

README

          

# Image-classifier
An AI model that uses OpenAI can classifier uploaded images

## Features
- GUI to upload the picture
- Supports JPG and PNG
- Uses OpenAI model
- Outputs 3 possible predicted classes and confidence score for each

## Requirements
- Python 3.8+
- `openai`, `numpy`, `streamlit`, `tensorflow`, `PIL`, `cv2`
- `.env` with OpenAI API key
- `uv` package

## Usage
- Install
- Create a `.env` file with your API key like this (`OPENAI_API_KEY=your_openai_api_key`)
- Run `main.py`

## Key insights
- Existing AI models like OpenAI eliminates the need to train new machine learning models from zero
- Steps to create a new AI model using OpenAI are very similar and straightforward