Save inference predictions
Fix
Fix