Update app.py

app.py

@@ -8,9 +8,12 @@ from dotenv import load_dotenv
 # Load environment variables
 load_dotenv()
 
-# Create a temporary cache directory
-writable_cache_dir = tempfile.mkdtemp()
-os.environ["TRANSFORMERS_CACHE"] = writable_cache_dir
+# Set the cache directory to the one specified in the environment variable
+cache_dir = os.getenv("TRANSFORMERS_CACHE", "/.cache/huggingface/hub")
+
+# Ensure the cache directory is writable
+os.makedirs(cache_dir, exist_ok=True)
+os.environ["TRANSFORMERS_CACHE"] = cache_dir
 
 # Retrieve the Hugging Face API token from environment variables
 api_token = os.getenv("ttt")