Spaces:
Running
Running
Update pages/13_FFNN.py
Browse files- pages/13_FFNN.py +1 -1
pages/13_FFNN.py
CHANGED
@@ -97,7 +97,7 @@ def imshow(img):
|
|
97 |
|
98 |
if st.sidebar.button('Show Test Results'):
|
99 |
dataiter = iter(testloader)
|
100 |
-
images, labels =
|
101 |
imshow(torchvision.utils.make_grid(images))
|
102 |
st.write('GroundTruth: ', ' '.join(f'{labels[j]}' for j in range(8)))
|
103 |
outputs = net(images)
|
|
|
97 |
|
98 |
if st.sidebar.button('Show Test Results'):
|
99 |
dataiter = iter(testloader)
|
100 |
+
images, labels = next(dataiter) # Use next function
|
101 |
imshow(torchvision.utils.make_grid(images))
|
102 |
st.write('GroundTruth: ', ' '.join(f'{labels[j]}' for j in range(8)))
|
103 |
outputs = net(images)
|