Fix Python Script

Hello, how are you doing? I hope you're doing well. I wrote a Python script to help in machine learning. The code contains a minor flaw. I want you to address this problem and plot the, model in picture, model's accuracy and loss as well. You are free to utilize any approach that is relevant to the task.