Fix feature extraction and models to match notebook

Initial backend deployment