A production-ready AI system that uses convolutional neural networks to predict whether a song has the characteristics of a hit or a flop.
- Real-time song prediction using Spotify's track data
- Interactive web interface
- Integration with Spotify's API
- Containerized application using Docker
- Neural network model trained on historical hit data
The system analyzes various musical features of a track (such as danceability, energy, tempo, etc.) using Spotify's audio features API and processes them through our trained neural network to predict its hit potential.
- Frontend: React, TypeScript, Tailwind CSS
- Backend: Python, FastAPI
- AI Model: TensorFlow, Keras
- Infrastructure: Docker, Docker Compose
The neural network model was trained using a comprehensive dataset of historical music data. You can find the training notebook and methodology on Kaggle: Spotify Hit or Flop Training Notebook
- Clone the repository
- Create a
.env
file with your Spotify API credentials - Run with Docker Compose: