Update main.py

main.py

@@ -9,6 +9,12 @@ import io
 from PIL import Image
 import os
 
+
+# Set custom Hugging Face cache directory
+os.environ["HF_HOME"] = "/app/cache"
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+os.environ["HF_HUB_CACHE"] = "/app/cache"
+
 # Initialize FastAPI app
 app = FastAPI(title="PromptAgro Image Generator API")