diff --git a/main.py b/main.py
index b52ac21..780e999 100644
--- a/main.py
+++ b/main.py
@@ -26,7 +26,8 @@ os.environ["OPENBLAS_NUM_THREADS"] = NUM_THREADS
 os.environ["MKL_NUM_THREADS"] = NUM_THREADS
 os.environ["VECLIB_MAXIMUM_THREADS"] = NUM_THREADS
 os.environ["NUMEXPR_NUM_THREADS"] = NUM_THREADS
-os.environ["TORCH_HOME"] = os.environ["cache_dir"]
+if os.environ["cache_dir"]:
+    os.environ["TORCH_HOME"] = os.environ["cache_dir"]
 
 BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "./lama_cleaner/app/build")