Hugging Face's logo

Spaces:
gremlin97
/
Tag2Text_Demo
Build error

App Files Community
gremlin97 commited on
Commit
bf02602
•
1 Parent(s): a7556f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import torchvision.transforms as transforms
10
  from PIL import Image
11
 
12
  evice = torch.device('cpu')
13
- model = tag2text(pretrained='./tag2text.pth', image_size=384, vit='swin_b')
14
  model.threshold = 0.68
15
  model.eval()
16
 
 
10
  from PIL import Image
11
 
12
  evice = torch.device('cpu')
13
+ model = tag2text(pretrained='./tag2text_swin_14m.pth', image_size=384, vit='swin_b')
14
  model.threshold = 0.68
15
  model.eval()
16