Fixed bug where model download link was saved instead of model page link.

2024-09-12 13:26:29 -04:00
parent 76c0ce817b
commit 15946599fb
1 changed files with 3 additions and 2 deletions
--- a/init.py
+++ b/init.py
@@ -1257,11 +1257,12 @@ async def download_model(request):
        return web.json_response(result)

    # download model info
-    _ = ModelInfo.search_info(file_name, cache=True) # save json
+    model_info = ModelInfo.search_info(file_name, cache=True) # save json

    # save url
    url_file_path = os.path.splitext(file_name)[0] + ".url"
-    save_web_url(url_file_path, download_uri)
+    url = ModelInfo.get_url(url)
+    save_web_url(url_file_path, url)

    # save image as model preview
    image = formdata.get("image")