debug: token.

main.py

@@ -34,13 +34,13 @@ model = tf.keras.applications.ResNet50(weights="imagenet")
 def push_to_hub(hf_token: str, push_dir: str) -> str:
     try:
         if hf_token is None:
-            return "No HF token provided. Model won't be pushed."
+            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 Exception as e:
-        return e
+            return f"### Model successfully pushed: [{commit_url}]({commit_url})"
+    except:
+        return "### Error happened."
 
 
 def post_optimization(list_of_strs: List[str]) -> str:
@@ -75,7 +75,7 @@ def run(hf_token: str) -> str:
     # benchmark_str += "\n"
     benchmark_str = push_to_hub(hf_token, tensorrt_path)
 
-    return f"### Token: {hf_token}"
+    return benchmark_str
 
 
 def launch_gradio():