delete cache between each run

app.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from tempfile import TemporaryDirectory
 
 import gradio as gr
-from huggingface_hub import HfApi, ModelCard
+from huggingface_hub import HfApi, ModelCard, scan_cache_dir
 from huggingface_hub.utils import EntryNotFoundError, RepositoryNotFoundError
 
 from convert import convert
@@ -129,12 +129,22 @@ def run(
             yield _log(
                 f"### Success 🔥\n\nYay! This model was successfully converted! Make sure to let the repo owner know about it and review your PR. You might need to complete the PR manually, especially to add information in the model card."
             )
+            _delete_cache()
             return
     except Exception as e:
         yield _log(f"### Error 😢😢😢\n\n{e}")
+        _delete_cache()
         return
 
 
+def _delete_cache():
+    """Delete cache dir between each run to avoid filling up the Space disk."""
+    scan = scan_cache_dir()
+    scan.delete_revisions(
+        *[rev.commit_hash for repo in scan.repos for rev in repo.revisions]
+    )
+
+
 TITLE = """
 <h1 style="font-weight: 900; font-size: 32px; margin-bottom: 10px; margin-top: 10px; text-align: center;">
     Make any BLOOM-like model compatible with bloomz.cpp
@@ -168,7 +178,9 @@ with gr.Blocks() as demo:
             gr.Markdown(DESCRIPTION)
 
         with gr.Column(scale=50):
-            input_token = gr.Text(max_lines=1, label="Hugging Face token", type="password")
+            input_token = gr.Text(
+                max_lines=1, label="Hugging Face token", type="password"
+            )
             input_model = gr.Text(
                 max_lines=1, label="Model id (e.g.: bigscience/bloomz-7b1)"
             )