Update main.py

main.py

@@ -22,16 +22,13 @@ logging.getLogger("uvicorn").setLevel(logging.WARNING)
 
 import tempfile
 
-# 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")
 
-# 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)
 
@@ -132,4 +129,4 @@ async def predict_cxr(image_url: str = Query(..., description="URL to a chest X-
         })
 
     except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
+        return JSONResponse(content={"error": str(e)}, status_code=500)