Update app.py

app.py

@@ -5,6 +5,7 @@ nltk.download('wordnet')
 from pytorch_pretrained_biggan import (BigGAN, one_hot_from_names, truncated_noise_sample,
                                        save_as_images, display_in_terminal)
 initial_archi = 'biggan-deep-128' #@param ['biggan-deep-128', 'biggan-deep-256', 'biggan-deep-512'] {allow-input: true}
+initial_class = 'dog'
 
 gan_model = BigGAN.from_pretrained(initial_archi)