up

app.py

@@ -30,17 +30,13 @@ def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken):
         start = time.perf_counter()
         compiled_transformer = compile_transformer(pipe, prompt="prompt")
         end = time.perf_counter()
-        print(f"Compilation took: {start - end} seconds.")
+        print(f"Compilation took: {end - start} seconds.")
 
         token = oauth_token.token
         out = _push_compiled_graph_to_hub(
             compiled_transformer.archive_file, repo_id=repo_id, token=token, path_in_repo=filename
         )
-        if not isinstance(out, str) and hasattr(out, "commit_url"):
-            commit_url = out.commit_url
-            return f"[{commit_url}]({commit_url})"
-        else:
-            return out
+        return out
     except Exception as e:
         raise gr.Error(
             f"""Oops, you forgot to login. Please use the loggin button on the top left to migrate your repo {e}"""

hub_utils.py

@@ -6,7 +6,7 @@ import os
 DEFAULT_ARCHIVE_FILENAME = "archived_graph.pt2"
 
 
-def _push_compiled_graph_to_hub(archive: BytesIO, repo_id, **kwargs):
+def _push_compiled_graph_to_hub(archive: BytesIO, repo_id, **kwargs) -> str:
     if not isinstance(archive, BytesIO):
         raise NotImplementedError("Incorrect type of `archive` provided.")
 
@@ -29,7 +29,7 @@ def _push_compiled_graph_to_hub(archive: BytesIO, repo_id, **kwargs):
                 commit_message=commit_message,
                 token=token,
             )
-            return info
+            return info.commit_url
         except Exception as e:
             print(f"File couldn't be pushed to the Hub with the following error: {e}.")
             return e