cache writing

main.py

@@ -7,8 +7,11 @@ import cv2
 from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 from PIL import Image
 import zipfile
-from transformers import pipeline
+
 import os
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
+
+from transformers import pipeline
 
 app = Flask(__name__)
 CORS(app)
@@ -16,7 +19,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"