alexfremont commited on
Commit
af74871
·
1 Parent(s): 38a3c61

fix cache issue

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -59,7 +59,7 @@ 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
+ cache_dir="./tmp/cache",
63
  )
64
  model = base_model.__class__("resnet152", num_output_neurons=2).to(device)
65
  model.load_state_dict(