AI tool to track virus variants