xinyu1205 commited on
Commit
79c9925
1 Parent(s): 542f1b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ transform = transforms.Compose([transforms.Resize((image_size, image_size)),tran
20
 
21
 
22
  #######Swin Version
23
- pretrained = '/home/notebook/code/personal/S9049611/BLIP/output/blip_tagtotext_14m/blip_tagtotext_encoderdiv_tar_random_swin/caption_coco_finetune_tagparse_tagfinetune_threshold075_bceloss_tagsingle_5e6_epoch19_negative_1_05_pos_1_10/checkpoint_05.pth'
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)