Update main.py

main.py

@@ -20,18 +20,20 @@ from io import BytesIO
 import logging
 logging.getLogger("uvicorn").setLevel(logging.WARNING)
 
-import os
-
-# Set a custom path for Matplotlib to use
-os.environ["MPLCONFIGDIR"] = "/app/.cache/matplotlib"
+import tempfile
 
-# Set a custom path for TorchXRayVision to use for model weights
-os.environ["TORCHXrayVISION_CACHE"] = "/app/.cache/torchxrayvision"
+# Instead of hardcoding paths to /app/.cache
+temp_dir = tempfile.gettempdir()
+matplotlib_cache = os.path.join(temp_dir, "matplotlib")
+torchxrayvision_cache = os.path.join(temp_dir, "torchxrayvision")
 
-# Create these directories if they do not exist
-os.makedirs("/app/.cache/matplotlib", exist_ok=True)
-os.makedirs("/app/.cache/torchxrayvision", exist_ok=True)
+# Set environment variables to use these paths
+os.environ["MPLCONFIGDIR"] = matplotlib_cache
+os.environ["TORCHXrayVISION_CACHE"] = torchxrayvision_cache
 
+# Create directories
+os.makedirs(matplotlib_cache, exist_ok=True)
+os.makedirs(torchxrayvision_cache, exist_ok=True)
 
 
 app = FastAPI()