End to End ML solution and deployment example
- Machine learning model to predict genre of movies based on the text overview
- Deployment using Docker
Steps to run:
- pip install -r requirements.txt
- python src/train.py
- python src/predict.py
- curl -d "overview=A movie about penguins in Antarctica building a spaceship to go to Mars." -X POST http://localhost:8005