chendl commited on
Commit
9acf8b9
1 Parent(s): e770d90

update cap

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,8 +53,8 @@ flamingo, image_processor, tokenizer, vis_embed_size = create_model_and_transfor
53
  )
54
 
55
 
56
- checkpoint_path = "/home/aimos/huggingface/space/demo.pt"
57
- # hf_hub_download("chendl/compositional_test", "pythiaS.pt")
58
  checkpoint = torch.load(checkpoint_path, map_location="cpu")["model_state_dict"]
59
  model_state_dict = {}
60
  for key in checkpoint.keys():
 
53
  )
54
 
55
 
56
+ # checkpoint_path = "/home/aimos/huggingface/space/demo.pt"
57
+ checkpoint_path = hf_hub_download("chendl/compositional_test", "pythiaS.pt")
58
  checkpoint = torch.load(checkpoint_path, map_location="cpu")["model_state_dict"]
59
  model_state_dict = {}
60
  for key in checkpoint.keys():