Update pages/23_GANs.py

pages/23_GANs.py

@@ -45,7 +45,6 @@ latent_dim = 100
 image_dim = 28 * 28  # MNIST images are 28x28 pixels
 lr = 0.0002
 batch_size = 64
-epochs = 50
 
 # Prepare the data
 transform = transforms.Compose([
@@ -71,6 +70,9 @@ criterion = nn.BCELoss()
 st.title("GAN with PyTorch and Hugging Face")
 st.write("Training a GAN to generate MNIST digits")
 
+# Slider for epochs
+epochs = st.slider("Number of Epochs", min_value=1, max_value=100, value=50)
+
 train_gan = st.button("Train GAN")
 
 if train_gan:
@@ -111,4 +113,4 @@ if train_gan:
     ax.imshow(np.squeeze(grid), cmap="gray")
     st.pyplot(fig)
 else:
-    st.write("Click the button to start training the GAN")
+    st.write("Use the slider to select the number of epochs and click the button to start training the GAN")