Spaces:
Running
Running
Upload 2 files
Browse files- README.md +1 -1
- civitai_to_hf.py +3 -3
README.md
CHANGED
|
@@ -4,7 +4,7 @@ emoji: 🤗
|
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: blue
|
| 6 |
sdk: gradio
|
| 7 |
-
sdk_version: 5.
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
|
|
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: blue
|
| 6 |
sdk: gradio
|
| 7 |
+
sdk_version: 5.31.0
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
civitai_to_hf.py
CHANGED
|
@@ -385,19 +385,19 @@ def get_civitai_json(dl_url: str, is_html: bool=False, image_baseurl: str="", ap
|
|
| 385 |
url = base_url + model_id
|
| 386 |
#url = base_url + str(dl_url.split("/")[-1])
|
| 387 |
try:
|
| 388 |
-
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0,
|
| 389 |
if not r.ok: return default
|
| 390 |
json = dict(r.json()).copy()
|
| 391 |
html = ""
|
| 392 |
image = ""
|
| 393 |
if "modelId" in json.keys():
|
| 394 |
url = f"https://civitai.com/models/{json['modelId']}"
|
| 395 |
-
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0,
|
| 396 |
if not r.ok: return json, html, image
|
| 397 |
html = r.text
|
| 398 |
if 'images' in json.keys() and len(json["images"]) != 0:
|
| 399 |
url = json["images"][0]["url"]
|
| 400 |
-
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0,
|
| 401 |
if not r.ok: return json, html, image
|
| 402 |
image_temp = str(Path(TEMP_DIR, "image" + Path(url.split("/")[-1]).suffix))
|
| 403 |
image = str(Path(TEMP_DIR, Path(image_baseurl.split("/")[-1]).stem + ".png"))
|
|
|
|
| 385 |
url = base_url + model_id
|
| 386 |
#url = base_url + str(dl_url.split("/")[-1])
|
| 387 |
try:
|
| 388 |
+
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0, 60))
|
| 389 |
if not r.ok: return default
|
| 390 |
json = dict(r.json()).copy()
|
| 391 |
html = ""
|
| 392 |
image = ""
|
| 393 |
if "modelId" in json.keys():
|
| 394 |
url = f"https://civitai.com/models/{json['modelId']}"
|
| 395 |
+
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0, 60))
|
| 396 |
if not r.ok: return json, html, image
|
| 397 |
html = r.text
|
| 398 |
if 'images' in json.keys() and len(json["images"]) != 0:
|
| 399 |
url = json["images"][0]["url"]
|
| 400 |
+
r = session.get(url, params=params, headers=headers, stream=True, timeout=(5.0, 60))
|
| 401 |
if not r.ok: return json, html, image
|
| 402 |
image_temp = str(Path(TEMP_DIR, "image" + Path(url.split("/")[-1]).suffix))
|
| 403 |
image = str(Path(TEMP_DIR, Path(image_baseurl.split("/")[-1]).stem + ".png"))
|