Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ transform = transforms.Compose([transforms.Resize((image_size, image_size)),tran
|
|
20 |
|
21 |
|
22 |
#######Swin Version
|
23 |
-
pretrained = '
|
24 |
|
25 |
config_file = 'configs/tag2text_caption.yaml'
|
26 |
config = yaml.load(open(config_file, 'r'), Loader=yaml.Loader)
|
|
|
20 |
|
21 |
|
22 |
#######Swin Version
|
23 |
+
pretrained = 'tag2text_swin_14m.pth'
|
24 |
|
25 |
config_file = 'configs/tag2text_caption.yaml'
|
26 |
config = yaml.load(open(config_file, 'r'), Loader=yaml.Loader)
|