Reproducing an AI model for Simulating 3D Transcranial Focused Ultrasound Stimulation