changes to app.py

app.py

@@ -3,6 +3,11 @@ os.environ['HF_HOME'] = '/tmp/hf_home'
 os.environ['HF_DATASETS_CACHE'] = '/tmp/hf_datasets_cache'
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/cache'
 
+# Create the directories and set permissions
+for d in ['/tmp/hf_home', '/tmp/hf_datasets_cache', '/tmp/cache']:
+    os.makedirs(d, exist_ok=True)
+    os.chmod(d, 0o777)
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM