Upload 2 files

app.py

@@ -14,9 +14,9 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_ca
                 search_civitai_basemodel = gr.CheckboxGroup(label="Base model", choices=CIVITAI_BASEMODEL, value=[])
             with gr.Row():
                 search_civitai_sort = gr.Radio(label="Sort", choices=CIVITAI_SORT, value=CIVITAI_SORT[0])
-                search_civitai_period = gr.Radio(label="Period", choices=CIVITAI_PERIOD, value=CIVITAI_PERIOD[0])
+                search_civitai_period = gr.Radio(label="Period", choices=CIVITAI_PERIOD, value="Month")
             with gr.Row():
-                search_civitai_query = gr.Textbox(label="Query", placeholder="oomuro sakurako...", lines=1)
+                search_civitai_query = gr.Textbox(label="Query", placeholder="flux", lines=1)
                 search_civitai_tag = gr.Textbox(label="Tag", lines=1)
                 search_civitai_submit = gr.Button("Search on Civitai")
             with gr.Row():

civitai_to_hf.py

@@ -86,12 +86,12 @@ def search_on_civitai(query: str, types: list[str], allow_model: list[str] = [],
                 item = {}
                 if len(allow_model) != 0 and model['baseModel'] not in set(allow_model): continue
                 item['name'] = j['name']
-                item['creator'] = j['creator']['username']
-                item['tags'] = j['tags']
-                item['model_name'] = model['name']
-                item['base_model'] = model['baseModel']
+                item['creator'] = j['creator']['username'] if 'creator' in j.keys() and 'username' in j['creator'].keys() else ""
+                item['tags'] = j['tags'] if 'tags' in j.keys() else ""
+                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>[LoRA Model URL](https://civitai.com/models/{j["id"]})'
+                item['md'] = f'<img src="{model["images"][0]["url"]}" alt="thumbnail" width="150" height="240"><br>[Model URL](https://civitai.com/models/{j["id"]})'
                 items.append(item)
         return items