Update predict_interaction.py