Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
162cbb0
1
Parent(s):
af74871
try fix cache issues
Browse files
main.py
CHANGED
@@ -59,7 +59,6 @@ async def load_models():
|
|
59 |
subfolder=model_location["subfolder"],
|
60 |
filename=model_location["filename"],
|
61 |
token=hf_token,
|
62 |
-
cache_dir="./tmp/cache",
|
63 |
)
|
64 |
model = base_model.__class__("resnet152", num_output_neurons=2).to(device)
|
65 |
model.load_state_dict(
|
|
|
59 |
subfolder=model_location["subfolder"],
|
60 |
filename=model_location["filename"],
|
61 |
token=hf_token,
|
|
|
62 |
)
|
63 |
model = base_model.__class__("resnet152", num_output_neurons=2).to(device)
|
64 |
model.load_state_dict(
|