Update app.py

app.py

@@ -18,7 +18,7 @@ normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
 transform = transforms.Compose([transforms.Resize((image_size, image_size)),transforms.ToTensor(),normalize])
 
 #######Tag2Text Model
-pretrained = '/home/notebook/data/group/huangxinyu/pretrain_model/tag2text/tag2text_swin_14m.pth'
+pretrained = 'tag2text_swin_14m.pth'
 
 model_tag2text = tag2text_caption(pretrained=pretrained, image_size=image_size, vit='swin_b' )
 
@@ -26,8 +26,8 @@ model_tag2text.eval()
 model_tag2text = model_tag2text.to(device)
 
 
-#######Swin Version
-pretrained = '/home/notebook/code/personal/S9049611/tag2text-v2/output/pretrain_tag2text_large_v2_14m_large_v14/new_coco_ori_finetune_384_v5_epoch03/checkpoint_01.pth'
+#######RAM Model
+pretrained = 'ram_swin_large_14m.pth'
 
 model_ram = ram(pretrained=pretrained, image_size=image_size, vit='swin_l' )