ondrejbiza commited on
Commit
04a8250
1 Parent(s): 911983c

Fix large download issue.

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")
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", 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)