Use local cache dir

model.py

@@ -25,7 +25,7 @@ class Model:
         }
         with keras.utils.custom_object_scope(custom_objects):
             weights = keras.models.load_model(
-                hf_hub_download(model_repo_id, model_filename, token=hf_token)
+                hf_hub_download(model_repo_id, model_filename, token=hf_token, cache_dir=".cache")
             ).get_weights()
             self.trained_model.set_weights(weights)