add clean up command and storage track

app.py

@@ -17,8 +17,12 @@ from cytof.utils import show_color_table
 
 # clean up accumulated tmp files
 import shutil
-shutil.rmtree("/home/user/.cache/huggingface", ignore_errors=True)
-shutil.rmtree("/home/user/.cache/torch", ignore_errors=True)
+shutil.rmtree("/home/user/.cache", ignore_errors=True)
+
+# prevent future cache growth
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
+os.environ["TORCH_HOME"] = "/tmp/torch"
 
 OUTDIR = './output'