Update app.py

app.py

@@ -49,14 +49,12 @@ device_type = 'cuda' if use_cuda else 'cpu'
 
 ### Captioning 
 config = 'configs/video/ePALM_video_caption_msrvtt.yaml'
-# config = yaml.load(open(config, 'r'), Loader=yaml.Loader)
 config = yaml.load(open(config, 'r'))
 
 text_model = 'facebook/opt-2.7b' 
-vision_model_name = 'vit_base_patch16_224'
+vision_model_name = 'timesformer'
+
 
-# text_model = 'facebook/opt-6.7b' 
-# vision_model_name = 'vit_large_patch16_224'
 
 start_layer_idx = 19
 end_layer_idx = 31