Upload civitai_to_hf.py

civitai_to_hf.py

@@ -91,7 +91,9 @@ def search_on_civitai(query: str, types: list[str], allow_model: list[str] = [],
                 item['model_name'] = model['name'] if 'name' in model.keys() else ""
                 item['base_model'] = model['baseModel'] if 'baseModel' in model.keys() else ""
                 item['dl_url'] = model['downloadUrl']
-                item['md'] = f'<img src="{model["images"][0]["url"]}" alt="thumbnail" width="150" height="240"><br>[Model URL](https://civitai.com/models/{j["id"]})'
+                if 'images' in model.keys():
+                    item['md'] = f'<img src="{model["images"][0]["url"]}" alt="thumbnail" width="150" height="240"><br>[Model URL](https://civitai.com/models/{j["id"]})'
+                else: item['md'] = f'[Model URL](https://civitai.com/models/{j["id"]})'
                 items.append(item)
         return items