1580158
/
sd-civitai-browser-plus
mirror of https://github.com/BlafKing/sd-civitai-browser-plus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Hotfix for CivitAI model card button

pull/245/head
BlafKing 2024-03-17 00:32:07 +01:00
parent 6877d1b60a
commit d2a638e59b
No known key found for this signature in database
GPG Key ID: BFD6B0BCA6280F8B
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/civitai_file_manage.py
View File

@ -417,18 +417,13 @@ def model_from_sent(model_name, content_type, tile_count):
output_html = _api.api_error_msg("offline")
modelID_failed = True
if not modelID_failed:
json_data = _api.api_to_data(content_type, "Newest", "AllTime", "Model name", None, None, None, tile_count, f"civitai.com/models/{modelID}")
json_data = _api.request_civit_api(f"https://civitai.com/api/v1/models?ids={modelID}")
else:
json_data = None
if json_data == "timeout":
output_html = _api.api_error_msg("timeout")
elif json_data == "error":
output_html = _api.api_error_msg("error")
elif json_data == "offline":
output_html = _api.api_error_msg("offline")
if isinstance(json_data, dict):
if not isinstance(json_data, dict):
output_html = _api.api_error_msg(json_data)
else:
model_versions = _api.update_model_versions(modelID, json_data)
output_html = _api.update_model_info(None, model_versions.get('value'), True, modelID, json_data, True)