Skip to content

Commit c70e3b9

Browse files
committed
fix: don't always report download as successful
1 parent d36bde1 commit c70e3b9

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

core.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -590,8 +590,10 @@ def run_tensorboard_script():
590590

591591
# Download
592592
def run_download_script(model_link: str):
593-
model_download_pipeline(model_link)
594-
return f"Model downloaded successfully."
593+
result = model_download_pipeline(model_link)
594+
if result == "Error" or result is None:
595+
return "An error occurred downloading the model. Check the console for details."
596+
return "Model downloaded successfully."
595597

596598

597599
# Prerequisites

0 commit comments

Comments
 (0)