A Python based Streamlit web app to detect AI-generated images
This Streamlit web app utilizes a custom-built convolutional neural network (CNN) that we developed and trained specifically for 10 distinct categories. The model analyzes image arrays to detect AI-generated content, providing detailed scores for each image.
- Airplane
- Automobile
- Bird
- Cat
- Deer
- Dog
- Frog
- Horse
- Ship
- Truck
- Detects AI-generated content in images
- Generate AI-images based on user prompt to test app
- Guessing game to interact with users
- Streamlit: To build to interactive web app.
- Python: Programming language.
- TensorFlow: Deep learning fask and image processing
- NumPy: Arrays for machine learning
- PyTorch:
- OS Module: Used to interact with file paths
- PIL: Image processing
- Hugging Face Hub: Pre-trained machine learning model and API