Spaces:
Running
Running
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)
|