This repository contains code and resources for an image classifier project. The project is built using a machine learning model to classify images as either AI-generated or real.
The image classifier project aims to distinguish between AI-generated images and real images using a pre-trained machine learning model. The project utilizes the Streamlit framework for creating an interactive web application.
To install the necessary dependencies, follow these steps:
- Clone this repository to your local machine.
- Navigate to the project directory.
- Install the dependencies listed in the
Requirements.txt
file:
pip install -r Requirements.txt
To run the image classifier application, use the following command:
streamlit run app.py
This will start the Streamlit application, and you can access it in your web browser.
Once the application is running, follow these steps:
- Click on the "Browse Files" button to select an image file (in JPG or PNG or JPEG or WEBM format).
- Wait for the model to load and process the image.
- The application will display the uploaded image along with the prediction result.
- The prediction will indicate whether the image is most likely AI-generated or a real image.