Going to need to have some way to associate 5000 with “large”. While 5000mAh is a reasonable phone battery is is not a large EV battery, now the model may have some training data that associates 5000 and phone with large, but if not, you need to make that connection by having perhaps added metadata, like when you see a capacity of 5000 you tag it with (large) giving you 5000(large)