Skip to content

Commit a55f922

Browse files
authored
Update check_install.py
1 parent 8e4701b commit a55f922

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dlclive/check_install/check_install.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,14 +53,14 @@ def main():
5353
# download dog test video from github:
5454
# TODO: Should check if the video's already there before downloading it (should have been cloned with the files)
5555
print(f"Downloading Video to {video_file}")
56-
url_link = "https://github.com/DeepLabCut/DeepLabCut-live/blob/master/check_install/dog_clip.avi?raw=True"
56+
url_link = "https://github.com/DeepLabCut/DeepLabCut-live/blob/main/check_install/dog_clip.avi?raw=True"
5757
urllib.request.urlretrieve(url_link, video_file, reporthook=urllib_pbar)
5858

5959
# download model from the DeepLabCut Model Zoo
6060
if Path(model_dir / SNAPSHOT_NAME).exists():
6161
print('Model already downloaded, using cached version')
6262
else:
63-
print("Downloading full_dog model from the DeepLabCut Model Zoo...")
63+
print("Downloading a test model from the DeepLabCut Model Zoo...")
6464
download_huggingface_model(MODEL_NAME, model_dir)
6565

6666
# assert these things exist so we can give informative error messages

0 commit comments

Comments
 (0)