Fix zero handling and single-digit inputs

Update model checkpoint with improved training for zero edge cases