Skip to content

Commit

Permalink
fix: model size to check for download
Browse files Browse the repository at this point in the history
  • Loading branch information
Kohulan committed Aug 30, 2023
1 parent f685841 commit b66bca6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion DECIMER/decimer.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@
# download models to a default location
if (
os.path.exists(model_path)
and os.stat(model_path + "/saved_model.pb").st_size != 28073658
and os.stat(model_path + "/saved_model.pb").st_size != 28080309
):
shutil.rmtree(model_path)
config.download_trained_weights(model_url, default_path)
Expand Down

0 comments on commit b66bca6

Please sign in to comment.