change python file to generate prediction graph