Update sorawm/iopaint/helper.py

sorawm/iopaint/helper.py

@@ -32,16 +32,18 @@ def switch_mps_device(model_name, device):
     return device
 
 
-def get_cache_path_by_url(url):
+def get_cache_path_by_url(url: str):
     parts = urlparse(url)
-    hub_dir = get_dir()
+    hub_dir = "/tmp/torch/hub"
     model_dir = os.path.join(hub_dir, "checkpoints")
     if not os.path.isdir(model_dir):
-        os.makedirs(model_dir)
+        os.makedirs(model_dir, exist_ok=True)
     filename = os.path.basename(parts.path)
     cached_file = os.path.join(model_dir, filename)
     return cached_file
 
+    
+
 
 def download_model(url, model_md5: str = None):
     if os.path.exists(url):