Yes, you can add AI to the drone using Raspberry Pi or any SBC to drone but you have to keep in mind the payload of drone. Here is steps how you can make a AI drone AI drone using Raspberry PI