transformer move cache

app.py

@@ -7,6 +7,10 @@ import spaces
 from train import main as train_main
 from inference import inference as inference_main
 
+import transformers
+transformers.utils.move_cache()
+
+
 @spaces.GPU()
 def inference_app(
         embedding_dir,