Update main.py

main.py

@@ -1,5 +1,6 @@
 import os
-os.environ["NUMBA_CACHE_DIR"] = "/tmp"       # Ensure a writable cache directory
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"  # Ensure the cache directory is writable
+os.environ["NUMBA_CACHE_DIR"] = "/tmp"       # Ensure a writable cache directory for Numba
 os.environ["NUMBA_DISABLE_CACHE"] = "1"        # Disable Numba caching to avoid errors
 
 import torch