Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 import nltk
 nltk.download('wordnet')
 nltk.download('omw-1.4')
-
+from PIL import Image
 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}