From 15946599fb58de7ce41e68bbbb5869ab39ef5686 Mon Sep 17 00:00:00 2001
From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com>
Date: Thu, 12 Sep 2024 13:26:29 -0400
Subject: [PATCH] Fixed bug where model download link was saved instead of
 model page link.

---
 __init__.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 06d4e2f..a64cc42 100644
--- 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")