Update app.py

app.py

@@ -4,6 +4,16 @@ from fastapi import FastAPI
 from transformers import pipeline
 
 pipe = pipeline("translation", model="google-t5/t5-base")
+import os
+
+# Set cache directory to a local writable directory
+os.environ['HF_HOME'] = './cache'
+os.environ['TRANSFORMERS_CACHE'] = './cache'
+os.environ['TORCH_HOME'] = './cache'
+
+# Ensure the directory exists
+os.makedirs('./cache', exist_ok=True)
+
 
 app = FastAPI()
 @app.get("/")