awacke1 commited on
Commit
00605e5
1 Parent(s): 92f4fef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,7 +16,7 @@ from PIL import Image
16
  from torchvision import transforms
17
 
18
  # === SEMI-WEAKLY SUPERVISED MODELSP RETRAINED WITH 940 HASHTAGGED PUBLIC CONTENT ===
19
- model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet18_swsl')
20
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet50_swsl')
21
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext50_32x4d_swsl')
22
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x4d_swsl')
@@ -32,8 +32,9 @@ model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 're
32
 
33
  io1 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
34
  io2 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
 
35
  #io3 = gr.Interface.load('vqgan_imagenet_f16_16384')
36
- io3 = gr.Interface.load(model)
37
  #io3 = gr.Interface.load("huggingface/emilyalsentzer/Bio_Discharge_Summary_BERT")
38
  #io3 = gr.Interface.load("huggingface/google/pegasus-pubmed")
39
  #io3 = gr.Interface.load("huggingface/tennessejoyce/titlewave-t5-base")
16
  from torchvision import transforms
17
 
18
  # === SEMI-WEAKLY SUPERVISED MODELSP RETRAINED WITH 940 HASHTAGGED PUBLIC CONTENT ===
19
+ #model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet18_swsl')
20
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnet50_swsl')
21
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext50_32x4d_swsl')
22
  # model = torch.hub.load('facebookresearch/semi-supervised-ImageNet1K-models', 'resnext101_32x4d_swsl')
32
 
33
  io1 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
34
  io2 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
35
+ io3 = gr.Interface.load('huggingface/osanseviero/BigGAN-deep-128')
36
  #io3 = gr.Interface.load('vqgan_imagenet_f16_16384')
37
+ #io3 = gr.Interface.load(model)
38
  #io3 = gr.Interface.load("huggingface/emilyalsentzer/Bio_Discharge_Summary_BERT")
39
  #io3 = gr.Interface.load("huggingface/google/pegasus-pubmed")
40
  #io3 = gr.Interface.load("huggingface/tennessejoyce/titlewave-t5-base")