diff --git a/scripts/civitai_api.py b/scripts/civitai_api.py index 0ddefd9..7143584 100644 --- a/scripts/civitai_api.py +++ b/scripts/civitai_api.py @@ -637,6 +637,9 @@ def update_model_info(model_string=None, model_version=None, only_html=False, in selected_version = model break + model_availability = selected_version.get('availability', 'Unknown') + model_date_published = selected_version.get('publishedAt', '').split('T')[0] + if selected_version['trainedWords']: output_training = ",".join(selected_version['trainedWords']) output_training = re.sub(r'<[^>]*:[^>]*>', '', output_training) @@ -818,6 +821,10 @@ def update_model_info(model_string=None, model_version=None, only_html=False, in