Spaces:

multimodalart
/

civitai-to-hf

Running

multimodalart HF Staff commited on Apr 11, 2024

Commit

e550a6d

verified ·

1 Parent(s): b1dc0ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ def extract_info(json_data):
                             "modelId": model_version.get("modelId", ""),
                             "name": json_data["name"],
                             "description": json_data["description"],
-                            "trainedWords": model_version["trainedWords"],
                             "creator": json_data["creator"]["username"],
                             "tags": json_data["tags"],
                             "allowNoCredit": json_data["allowNoCredit"],

                             "modelId": model_version.get("modelId", ""),
                             "name": json_data["name"],
                             "description": json_data["description"],
+                            "trainedWords": model_version["trainedWords"] if "trainedWords" in model_version else [],
                             "creator": json_data["creator"]["username"],
                             "tags": json_data["tags"],
                             "allowNoCredit": json_data["allowNoCredit"],