Spaces:
Running
Running
Upload civitai_to_hf.py
Browse files- civitai_to_hf.py +1 -1
civitai_to_hf.py
CHANGED
@@ -87,7 +87,7 @@ def search_on_civitai(query: str, types: list[str], allow_model: list[str] = [],
|
|
87 |
if len(allow_model) != 0 and model['baseModel'] not in set(allow_model): continue
|
88 |
item['name'] = j['name']
|
89 |
item['creator'] = j['creator']['username'] if 'creator' in j.keys() and 'username' in j['creator'].keys() else ""
|
90 |
-
item['tags'] = j['tags'] if 'tags' in j.keys() else
|
91 |
item['model_name'] = model['name'] if 'name' in model.keys() else ""
|
92 |
item['base_model'] = model['baseModel'] if 'baseModel' in model.keys() else ""
|
93 |
item['dl_url'] = model['downloadUrl']
|
|
|
87 |
if len(allow_model) != 0 and model['baseModel'] not in set(allow_model): continue
|
88 |
item['name'] = j['name']
|
89 |
item['creator'] = j['creator']['username'] if 'creator' in j.keys() and 'username' in j['creator'].keys() else ""
|
90 |
+
item['tags'] = j['tags'] if 'tags' in j.keys() else []
|
91 |
item['model_name'] = model['name'] if 'name' in model.keys() else ""
|
92 |
item['base_model'] = model['baseModel'] if 'baseModel' in model.keys() else ""
|
93 |
item['dl_url'] = model['downloadUrl']
|