cleanup cache

app.py

@@ -2,6 +2,7 @@
 
 import gradio as gr
 import os
+import shutil
 import torch
 
 from huggingface_hub import hf_hub_download
@@ -18,7 +19,7 @@ def download_model(cfg):
         repo_id=cfg.repo_id, 
         filename=cfg.model_name, 
         token=HF_TOKEN,
-        cache_dir="./model-cache"
+        cache_dir=cfg.cache_dir
     )
     return model_path
 
@@ -66,6 +67,7 @@ if __name__ == '__main__':
         'model_type': 'transformer',
         'repo_id': "Borzyszkowski/AlpineLLM-model",
         'model_name': "best_model",
+        'cache_dir': "./model-cache",
     }
     cfg = Config(cfg)
 
@@ -81,6 +83,7 @@ if __name__ == '__main__':
     hyperparam_cfg = Config(hyperparam_cfg)
 
     # Ensure model weights are available
+    shutil.rmtree(cfg.cache_dir, ignore_errors=True)
     cfg.load_weights_path = download_model(cfg)
 
     # Start the application