Hello dear aspirant.

Yes you can surely choose computer science specialisation in artificial intelligence after completing diploma in EEE. But it is going to be a little tough for you so I advise you that if you are highly interested in it then only opt for it. This way you will surely succeed else some problems might be there.

Good luck