Hugging Face's logo

Spaces:
ondrejbiza
/
isa
Running

App Files Community
ondrejbiza commited on
Commit
1ccf223
•
1 Parent(s): 04a8250

Fix V2.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def load_image(name):
61
  return img
62
 
63
 
64
- download_path = snapshot_download(repo_id="ondrejbiza/isa", filename="clevr_isa_ts")
65
  checkpoint_dir = os.path.join(download_path, "clevr_isa_ts")
66
 
67
  model, state, rng = load_model(get_config(), checkpoint_dir)
61
  return img
62
 
63
 
64
+ download_path = snapshot_download(repo_id="ondrejbiza/isa", allow_patterns="clevr_isa_ts*")
65
  checkpoint_dir = os.path.join(download_path, "clevr_isa_ts")
66
 
67
  model, state, rng = load_model(get_config(), checkpoint_dir)