refactored code

main.py

@@ -1,4 +1,10 @@
 import os
+
+# Set Hugging Face cache directory to /tmp
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+os.environ["TORCH_HOME"] = "/tmp/torch"
+
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import Response
@@ -9,11 +15,6 @@ import numpy as np
 from lang_sam import LangSAM
 import supervision as sv
 
-# Set Hugging Face cache directory to /tmp
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
-os.environ["TORCH_HOME"] = "/tmp/torch"
-
 app = FastAPI()
 
 # Enable CORS for all origins