Update app.py

app.py

@@ -1,4 +1,3 @@
-import ruamel_yaml as yaml
 import numpy as np
 import random
 
@@ -22,12 +21,7 @@ transform = transforms.Compose([transforms.Resize((image_size, image_size)),tran
 #######Swin Version
 pretrained = 'tag2text_swin_14m.pth'
 
-config_file = 'configs/tag2text_caption.yaml'
-config = yaml.load(open(config_file, 'r'), Loader=yaml.Loader)
-
-model = tag2text_caption(pretrained=pretrained, image_size=image_size, vit=config['vit'], 
-                    vit_grad_ckpt=config['vit_grad_ckpt'], vit_ckpt_layer=config['vit_ckpt_layer'],
-                    prompt=config['prompt'],config=config,threshold = 0.75 )
+model = tag2text_caption(pretrained=pretrained, image_size=image_size, vit='swin_b' )
 
 model.eval()
 model = model.to(device)