Update pages/1_Simple_CNN.py

pages/1_Simple_CNN.py

@@ -92,7 +92,7 @@ def train_model(num_epochs):
         plt.show()
 
     dataiter = iter(testloader)
-    images, labels = dataiter.next()
+    images, labels = next(dataiter)
 
     imshow(torchvision.utils.make_grid(images))
 
@@ -108,3 +108,4 @@ st.title('CIFAR-10 Classification with PyTorch')
 num_epochs = st.number_input('Enter number of epochs:', min_value=1, max_value=100, value=10)
 if st.button('Run'):
     train_model(num_epochs)
+