Update app.py

app.py

@@ -9,6 +9,13 @@ from dotenv import load_dotenv
 from src.prompt import *
 import os
 
+import os
+
+# Set custom cache directories
+os.environ['TRANSFORMERS_CACHE'] = '/app/model_cache'
+os.environ['HF_HOME'] = '/app/model_cache'
+os.makedirs('/app/model_cache', exist_ok=True)
+
 app = Flask(__name__)
 
 # Load environment variables - these will be set in Hugging Face Space secrets