Update app.py

app.py

@@ -55,6 +55,13 @@ async def load_model():
     
     print("🚀 Starting Text Correction API...")
     
+    # Set cache directory if not already set
+    import os
+    if not os.environ.get("TRANSFORMERS_CACHE"):
+        os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+    if not os.environ.get("HF_HOME"):
+        os.environ["HF_HOME"] = "/app/cache"
+    
     # Determine device
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"📱 Using device: {device}")