Update app.py

app.py

@@ -33,6 +33,22 @@ for dir_path in [CACHE_DIR, UPLOAD_DIR]:
 os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
 os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
 
+for dir_path in [CACHE_DIR, UPLOAD_DIR]:
+    Path(dir_path).mkdir(parents=True, exist_ok=True)
+
+# Set environment variables
+os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
+os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
+
+# Add some debugging to verify the environment variables
+print(f"Transformers Cache Directory: {os.environ['TRANSFORMERS_CACHE']}")
+print(f"Hugging Face Hub Cache Directory: {os.environ['HUGGINGFACE_HUB_CACHE']}")
+
+# Check if cache directories are writable
+if os.access(CACHE_DIR, os.W_OK):
+    print(f"Cache directory {CACHE_DIR} is writable.")
+else:
+    print(f"Cache directory {CACHE_DIR} is NOT writable.")
 # Configure Flask app
 app.config['UPLOAD_FOLDER'] = UPLOAD_DIR