Fix

app_upload.py

@@ -21,6 +21,7 @@ class ModelUploader(Uploader):
         private: bool,
         delete_existing_repo: bool,
         input_hf_token: str | None = None,
+        return_html_link: bool = True,
     ) -> str:
         if not folder_path:
             raise ValueError
@@ -40,7 +41,8 @@ class ModelUploader(Uploader):
                            organization=organization,
                            private=private,
                            delete_existing_repo=delete_existing_repo,
-                           input_hf_token=input_hf_token)
+                           input_hf_token=input_hf_token,
+                           return_html_link=return_html_link)
 
 
 def load_local_model_list() -> dict:

trainer.py

@@ -152,7 +152,8 @@ class Trainer:
                 upload_to=upload_to,
                 private=use_private_repo,
                 delete_existing_repo=delete_existing_repo,
-                input_hf_token=input_hf_token)
+                input_hf_token=input_hf_token,
+                return_html_link=False)
             with open(self.log_file, 'a') as f:
                 f.write(upload_message)
 

uploader.py

@@ -14,7 +14,8 @@ class Uploader:
                repo_type: str = 'model',
                private: bool = True,
                delete_existing_repo: bool = False,
-               input_hf_token: str | None = None) -> str:
+               input_hf_token: str | None = None,
+               return_html_link: bool = True) -> str:
 
         api = HfApi(token=self.hf_token or input_hf_token)
 
@@ -38,7 +39,9 @@ class Uploader:
                               path_in_repo='.',
                               repo_type=repo_type)
             url = f'https://huggingface.co/{repo_id}'
-            message = f'Your model was successfully uploaded to <a href="{url}" target="_blank">{url}</a>.'
+            if return_html_link:
+                url = f'<a href="{url}" target="_blank">{url}</a>'
+            message = f'Your model was successfully uploaded to {url}.'
         except Exception as e:
             message = str(e)
         return message