This streamlit based Web app checks if a image is AI generated or Real. We trained VGG16 model on 300 X 300 X 3 dimensional 9000 AI generated and 5000 Real images using feature extraction transfer learning to classify images into two categories(AI generated, Real).
To install the necessary dependencies, follow these steps:
- Clone this repository in your machine.
- Navigate to the project directory.
- Install all 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 classifier.py
This will start the Streamlit application on localhost, and you can access it in your web browser.
Once web app is running on your localhost:
- Select an image by clicking on the "Browse Files" button.
- Model will predict image category.
- The application will display the uploaded image along with the prediction result.