Merge pull request #13 from DrStalker/main

Add the URL of the Civitai model page to output
2024-06-20 17:29:46 +05:00
parent b4e88e4664 26c0f32929
commit 7a3becba72
1 changed files with 4 additions and 1 deletions
@@ -74,7 +74,10 @@ def get_lora_info(lora_name):
        baseModel = model_info.get("baseModel", "")
        images = model_info.get('images')
        examplePrompt = None
-
+        modelID = model_info.get("modelId")
+        
+        if modelID:
+            output += f"URL: https://civitai.com/models/{modelID}\n"
        if trainedWords:
            output += "Triggers: " + trainedWords
            output += "\n"