Update m5.py

m5.py

@@ -5,6 +5,8 @@ from accelerate import Accelerator
 from transformers import AutoTokenizer, AutoModel
 from sklearn.metrics.pairwise import cosine_similarity
 
+# Set the TRANSFORMERS_CACHE environment variable to a writable directory
+os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 class DM(nn.Module):