Set cache directory for LangSAM

main.py

@@ -1,3 +1,6 @@
+import os
+os.environ["TORCH_HOME"] = "/app/cache"  # Set a writable directory
+
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import Response
@@ -10,15 +13,18 @@ import supervision as sv
 
 app = FastAPI()
 
-# Enable CORS for all origins (Adjust as needed)
+# Enable CORS for all origins
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allow requests from any origin (Change this for security)
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Allow all HTTP methods
-    allow_headers=["*"],  # Allow all headers
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
+# Create a cache directory in the app space
+os.makedirs("/app/cache", exist_ok=True)
+
 # Load the segmentation model
 model = LangSAM()