John6666 commited on
Commit
95cf603
1 Parent(s): f7f6d75

Upload civitai_to_hf.py

Browse files
Files changed (1) hide show
  1. 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']