Upload helper.py

iopaint/helper.py

@@ -33,10 +33,16 @@ def switch_mps_device(model_name, device):
 
 def get_cache_path_by_url(url):
     parts = urlparse(url)
-    hub_dir = get_dir()
-    model_dir = os.path.join(hub_dir, "checkpoints")
-    if not os.path.isdir(model_dir):
-        os.makedirs(model_dir)
+    try:
+        model_dir = "/app/models/checkpoints"
+        os.makedirs(model_dir, exist_ok=True)
+        logger.info(f"Using model directory: {model_dir}")
+    except Exception as e:
+        logger.warning(f"Failed to create /app/models/checkpoints: {e}")
+        model_dir = "/tmp/iopaint/checkpoints"
+        os.makedirs(model_dir, exist_ok=True)
+        logger.info(f"Using alternative model directory: {model_dir}")
+    
     filename = os.path.basename(parts.path)
     cached_file = os.path.join(model_dir, filename)
     return cached_file