Spaces:
Running
Running
Upload utils.py
Browse files
utils.py
CHANGED
@@ -130,7 +130,7 @@ def download_thing(directory, url, civitai_api_key="", progress=gr.Progress(trac
|
|
130 |
if "drive.google.com" in url:
|
131 |
original_dir = os.getcwd()
|
132 |
os.chdir(directory)
|
133 |
-
subprocess.run(f"gdown --fuzzy
|
134 |
os.chdir(original_dir)
|
135 |
elif "huggingface.co" in url:
|
136 |
url = url.replace("?download=true", "")
|
@@ -138,13 +138,13 @@ def download_thing(directory, url, civitai_api_key="", progress=gr.Progress(trac
|
|
138 |
download_hf_file(directory, url)
|
139 |
elif "civitai.com" in url:
|
140 |
if civitai_api_key:
|
141 |
-
url =
|
142 |
print(f"Downloading {url}")
|
143 |
-
subprocess.run(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory}
|
144 |
else:
|
145 |
print("You need an API key to download Civitai models.")
|
146 |
else:
|
147 |
-
os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory}
|
148 |
except Exception as e:
|
149 |
print(f"Failed to download: {e}")
|
150 |
|
|
|
130 |
if "drive.google.com" in url:
|
131 |
original_dir = os.getcwd()
|
132 |
os.chdir(directory)
|
133 |
+
subprocess.run(f"gdown --fuzzy {url}", shell=True)
|
134 |
os.chdir(original_dir)
|
135 |
elif "huggingface.co" in url:
|
136 |
url = url.replace("?download=true", "")
|
|
|
138 |
download_hf_file(directory, url)
|
139 |
elif "civitai.com" in url:
|
140 |
if civitai_api_key:
|
141 |
+
url = f"'{url}&token={civitai_api_key}'" if "?" in url else f"{url}?token={civitai_api_key}"
|
142 |
print(f"Downloading {url}")
|
143 |
+
subprocess.run(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}", shell=True)
|
144 |
else:
|
145 |
print("You need an API key to download Civitai models.")
|
146 |
else:
|
147 |
+
os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
|
148 |
except Exception as e:
|
149 |
print(f"Failed to download: {e}")
|
150 |
|