Upload folder using huggingface_hub

demo.py

@@ -26,6 +26,7 @@ parser.add_argument("--local", action='store_true')
 args = parser.parse_args()
 
 if not args.local:
+    print("=> Logging into huggingface...")
     from huggingface_hub import login
     login(token=os.environ["HF_TOKEN"])
 
@@ -45,7 +46,8 @@ model = create_model(f"tpmlp_{size}").to(device)
 try:
     load_checkpoint(model, f"../tpmlp_{size}.pth.tar", True)
 except FileNotFoundError:
-    os.system(f"wget https://f005.backblazeb2.com/file/cs-mixer/tpmlp_{size}.pth.tar")
+    print(f"Downloading checkpoint tpmlp_{size}.")
+    os.system(f"wget https://f005.backblazeb2.com/file/cs-mixer/tpmlp_{size}.pth.tar &> /dev/null")
     load_checkpoint(model, f"tpmlp_{size}.pth.tar", True)
 model.eval()