AI could help fight the spread infectious disease around the world