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")
|