Fix: Use /tmp for caching

main.py

@@ -1,15 +1,12 @@
 import os
-os.environ["TORCH_HOME"] = "/app/cache"  # Set a writable directory
 
-from fastapi import FastAPI, File, UploadFile, Form
+# Set the cache directory to /tmp (which is writable in Hugging Face Spaces)
+os.environ["TORCH_HOME"] = "/tmp/cache"
+
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import Response
 import uvicorn
-from PIL import Image
-import io
-import numpy as np
 from lang_sam import LangSAM
-import supervision as sv
 
 app = FastAPI()
 
@@ -22,12 +19,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Create a cache directory in the app space
-os.makedirs("/app/cache", exist_ok=True)
+# Create a cache directory in /tmp
+os.makedirs("/tmp/cache", exist_ok=True)
 
 # Load the segmentation model
 model = LangSAM()
 
+@app.get("/")
+async def root():
+    return {"message": "LangSAM API is running!"}
+
 def draw_image(image_rgb, masks, xyxy, probs, labels):
     mask_annotator = sv.MaskAnnotator()
     # Create class_id for each unique label
@@ -72,3 +73,7 @@ async def segment_image(file: UploadFile = File(...), text_prompt: str = Form(..
     img_io.seek(0)
     
     return Response(content=img_io.getvalue(), media_type="image/png")
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)