visual changes to app.py

app.py

@@ -12,12 +12,9 @@ if torch.cuda.is_available():
     device = torch.device("cuda")
 
 latent_size = 100
-display_width = 450
 checkpoint_path = "Checkpoints/150epochs.chkpt"
 
 st.title("Generating Abstract Art")
-st.text("start generating (left side bar)")
-st.text("Made by Xingyu B.")
 
 st.sidebar.subheader("Configurations")
 seed = st.sidebar.slider('Seed', -10000, 10000, 0)
@@ -59,7 +56,6 @@ if generate:
     # use srgan for super resolution
     if use_srgan == "Yes":
         # restore to the checkpoint
-        st.write("Using DCGAN then ESRGAN upscale...")
         esrgan_generator = SRGAN.GeneratorRRDB(channels=3, filters=64, num_res_blocks=23).to(device)
         esrgan_checkpoint = load_esrgan()
         esrgan_generator.load_state_dict(esrgan_checkpoint)
@@ -71,15 +67,14 @@ if generate:
 
         for i in range(len(color_match)):
             # denormalize and permute to correct color channel
-            st.image(denormalize_images(color_match[i]).permute(1, 2, 0).numpy(), width=display_width)
+            st.image(denormalize_images(color_match[i]).permute(1, 2, 0).numpy())
 
 
     # default setting -> vanilla dcgan generation
     if use_srgan == "No":
         fakes = fakes.cpu()
-        st.write("Using DCGAN Model...")
         for i in range(len(fakes)):
-            st.image(denormalize_images(fakes[i]).permute(1, 2, 0).numpy(), width=display_width)
+            st.image(denormalize_images(fakes[i]).permute(1, 2, 0).numpy())