cleanup.py

import os
import shutil

def free_up_space():
    # Check disk usage before cleanup
    print("Disk usage before cleanup:")
    os.system("df -h")

    # Clear Hugging Face cache
    print("Clearing Hugging Face cache...")
    shutil.rmtree(os.path.expanduser("~/.cache/huggingface"), ignore_errors=True)
    shutil.rmtree(os.path.expanduser("~/.cache/transformers"), ignore_errors=True)

    # Clear PyTorch cache
    print("Clearing PyTorch cache...")
    shutil.rmtree(os.path.expanduser("~/.cache/torch"), ignore_errors=True)

    # Check disk usage after cleanup
    print("Disk usage after cleanup:")
    os.system("df -h")

if __name__ == "__main__":
    free_up_space()