Modify the deprecated code in the main.py file of the pytorch_basics repository.

[Corner430]

The line images, labels = data_iter.next() has been deprecated. I have updated it to images, labels = next(data_iter).

Additionally, I've split the main.py file into a new file named main.ipynb.