Spaces:
Runtime error
Runtime error
update cap
Browse files
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 |
-
|
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():
|