Update uploader.py

uploader.py

@@ -5,10 +5,7 @@ from huggingface_hub import HfApi
 
 class Uploader:
     def __init__(self, hf_token: str | None):
-        self.api = HfApi(token=hf_token)
-
-    def get_username(self) -> str:
-        return self.api.whoami()['name']
+        self.hf_token = hf_token
 
     def upload(self,
                folder_path: str,
@@ -16,13 +13,18 @@ class Uploader:
                organization: str = '',
                repo_type: str = 'model',
                private: bool = True,
-               delete_existing_repo: bool = False) -> str:
+               delete_existing_repo: bool = False,
+               input_token: str | None = False) -> str:
+        
+        api = HfAPi(token=self.hf_token if self.hf_token else input_token)
+        
         if not folder_path:
             raise ValueError
         if not repo_name:
             raise ValueError
         if not organization:
-            organization = self.get_username()
+            organization = api.whoami()['name']
+        
         repo_id = f'{organization}/{repo_name}'
         if delete_existing_repo:
             try:
@@ -30,11 +32,13 @@ class Uploader:
             except Exception:
                 pass
         try:
-            self.api.create_repo(repo_id, repo_type=repo_type, private=private)
-            self.api.upload_folder(repo_id=repo_id,
-                                   folder_path=folder_path,
-                                   path_in_repo='.',
-                                   repo_type=repo_type)
+            api = HfAPi(token=self.hf_token if self.hf_token else input_token)
+                
+            api.create_repo(repo_id, repo_type=repo_type, private=private)
+            api.upload_folder(repo_id=repo_id,
+                              folder_path=folder_path,
+                              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>.'
         except Exception as e: