test cache

main.py

@@ -8,6 +8,7 @@ from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 from PIL import Image
 import zipfile
 from transformers import pipeline
+import os
 
 app = Flask(__name__)
 CORS(app)
@@ -15,6 +16,7 @@ CORS(app)
 cudaOrNah = "cuda" if torch.cuda.is_available() else "cpu"
 print(cudaOrNah)
 
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
 # Global model setup 
 # running out of memory adjusted
 # checkpoint = "sam_vit_h_4b8939.pth"