Update app.py

app.py

@@ -10,10 +10,30 @@ 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"}
 
+import os
+import torch
+from PIL import Image
+from torchvision import transforms
+
+# === SEMI-WEAKLY SUPERVISED MODELSP RETRAINED WITH 940 HASHTAGGED PUBLIC CONTENT ===
+model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet18_swsl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet50_swsl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext50_32x4d_swsl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x4d_swsl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x8d_swsl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x16d_swsl')
+# ================= SEMI-SUPERVISED MODELS PRETRAINED WITH YFCC100M ==================
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet18_ssl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet50_ssl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext50_32x4d_ssl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x4d_ssl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x8d_ssl')
+# model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x16d_ssl')
 
 io1 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
 io2 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
-io3 = gr.Interface.load('vqgan_imagenet_f16_16384')                  
+#io3 = gr.Interface.load('vqgan_imagenet_f16_16384')                  
+io3 = gr.Interface.load(model)                  
 #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")