Upload folder using huggingface_hub
Browse files
demo.py
CHANGED
@@ -35,7 +35,11 @@ IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
|
|
35 |
IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
|
36 |
|
37 |
model = create_model(f"tpmlp_{size}").to(device)
|
38 |
-
|
|
|
|
|
|
|
|
|
39 |
model.eval()
|
40 |
|
41 |
response = requests.get("https://git.io/JJkYN")
|
|
|
35 |
IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
|
36 |
|
37 |
model = create_model(f"tpmlp_{size}").to(device)
|
38 |
+
try:
|
39 |
+
load_checkpoint(model, f"../tpmlp_{size}.pth.tar", True)
|
40 |
+
except FileNotFoundError:
|
41 |
+
os.system(f"wget https://f005.backblazeb2.com/file/cs-mixer/tpmlp_{size}.pth.tar")
|
42 |
+
load_checkpoint(model, f"tpmlp_{size}.pth.tar", True)
|
43 |
model.eval()
|
44 |
|
45 |
response = requests.get("https://git.io/JJkYN")
|