Update app.py

app.py

@@ -8,9 +8,13 @@ title = "BigGAN ImageNet"
 
 Gans=["biggan-deep-128", "biggan-deep-256", "biggan-deep-512"]
 
+    model_names={"vqgan_imagenet_f16_16384": 'ImageNet 16384',"vqgan_imagenet_f16_1024":"ImageNet 1024", 'vqgan_openimages_f16_8192':'OpenImages 8912',
+                    "wikiart_1024":"WikiArt 1024", "wikiart_16384":"WikiArt 16384", "coco":"COCO-Stuff", "faceshq":"FacesHQ", "sflckr":"S-FLCKR"}
+
+
 io1 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
 io2 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
-io3 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')                  
+io3 = gr.Interface.load('vqgan_imagenet_f16_16384')                  
 #io3 = gr.Interface.load("huggingface/emilyalsentzer/Bio_Discharge_Summary_BERT")                  
 #io3 = gr.Interface.load("huggingface/google/pegasus-pubmed")  
 #io3 = gr.Interface.load("huggingface/tennessejoyce/titlewave-t5-base")