fix: hub push.

main.py

@@ -2,7 +2,7 @@ from typing import List
 
 import gradio as gr
 import tensorflow as tf
-from huggingface_hub import Repository
+from huggingface_hub import HfApi, Repository, create_repo
 
 from utils import benchmark, convert_to_trt
 
@@ -36,11 +36,14 @@ def push_to_hub(hf_token: str, push_dir: str) -> str:
         if hf_token == "":
             return "### No HF token provided. Model won't be pushed."
         else:
-            repo = Repository(local_dir=push_dir, token=hf_token)
-            commit_url = repo.push_to_hub()
-            return f"### Model successfully pushed: [{commit_url}]({commit_url})"
-    except:
-        return f"### {hf_token == ''}."
+            hf_api = HfApi(token=hf_token)
+            user = hf_api.whoami()["name"]
+            repo_id = f"{user}/{push_dir}"
+            _ = create_repo(repo_id=repo_id, token=hf_token)
+            url = hf_api.upload_folder(folder_path=push_dir, repo_id=repo_id)
+            return f"### Model successfully pushed: [{url}]({url})"
+    except Exception as e:
+        return f"### {e}"
 
 
 def post_optimization(list_of_strs: List[str]) -> str: