customize cache dir

run.py

@@ -8,8 +8,9 @@ import os
 def pred(src):
     # os.environ['HUGGINGFACE_HUB_HOME'] = './.cache'
     # Load the model configuration
-    model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
-    processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
+    cache_dir = "/code/cache"
+    model_config = SamConfig.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
+    processor = SamProcessor.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
 
     # Create an instance of the model architecture with the loaded configuration
     my_sam_model = SamModel(config=model_config)