Update main.py

main.py

@@ -12,6 +12,10 @@ from io import BytesIO
 from PIL import Image
 import requests
 import os
+import os
+
+# Set the Hugging Face cache directory to a writable location
+os.environ['HF_HOME'] = '/tmp/huggingface_cache'
 
 app = FastAPI()