Skip to content

Commit af7fb43

Browse files
committed
feat: edit downloader module, make it simpler
1 parent 675454e commit af7fb43

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/onepiece_classify/utils/downloader.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,8 @@ def downloader(output_file):
77
prefix = 'https://drive.google.com/uc?/export=download&id='
88

99
url_download = prefix+file_id
10-
filename = "checkpoint_notebook.pth"
11-
if Path(output_file).joinpath(filename).exists():
12-
print("The model has been downloaded")
13-
else:
10+
11+
if not Path(output_file).exists():
1412
print("Downloading...")
1513
gdown.download(url_download, output_file)
1614
print("Download Finish...")

0 commit comments

Comments
 (0)