Few tweaks (#7)

- Few tweaks (9d9eebf0b35035d8b69306fb376f45136a1ba60b)


Co-authored-by: Lucain Pouget <[email protected]>

app.py

@@ -1,7 +1,7 @@
 import os
 import time
 os.system("wget https://raw.githubusercontent.com/Weyaxi/scrape-open-llm-leaderboard/main/openllm.py")
-from huggingface_hub import CommitOperationAdd, create_commit, HfApi, HfFileSystem, login
+from huggingface_hub import CommitOperationAdd, create_commit, HfApi, HfFileSystem, RepoUrl
 from huggingface_hub import ModelCardData, EvalResult, ModelCard
 from huggingface_hub.repocard_data import eval_results_to_model_index
 from huggingface_hub.repocard import RepoCard
@@ -166,6 +166,12 @@ def commit(repo, pr_number=None, message="Adding Evaluation Results", oauth_toke
   if oauth_token.expires_at < time.time():
     raise gr.Error("Token expired. Logout and try again.")
   token = oauth_token.token
+
+  if repo.startswith("https://huggingface.co/"):
+      try:
+        repo = RepoUrl(repo).repo_id
+      except Exception:
+        raise gr.Error(f"Not a valid repo id: {str(repo)}")
     
   edited = {"revision": f"refs/pr/{pr_number}"} if pr_number else {"create_pr": True}
 
@@ -216,24 +222,19 @@ The leaderboard's backend mainly runs on the [Hugging Face Hub API](https://hugg
 """
 
 with gr.Blocks() as demo:
-    gr.Markdown(f"""<h1 align="center" id="space-title">{gradio_title}</h1>""")
+    gr.HTML(f"""<h1 align="center" id="space-title">{gradio_title}</h1>""")
     gr.Markdown(gradio_desc)
 
     with gr.Row(equal_height=False):
         with gr.Column():
-            space_id = gr.Textbox(label="Model ID or URL", lines=1)
+            model_id = gr.Textbox(label="Model ID or URL", lines=1)
             gr.LoginButton()
 
         with gr.Column():
             output = gr.Textbox(label="Output", lines=1)
             gr.LogoutButton()
 
-    submit_btn = gr.Button("Submit", variant="primary")
-    try:
-        space_id = "/".join(space_id.split('/')[-2::]) if space_id.startswith("https://huggingface.co/") else space_id
-    except Exception as e:
-        gr.Error(f"There is an error with your model ID, check it again: {e}")
-        
-    submit_btn.click(commit, space_id, output)
+    submit_btn = gr.Button("Submit", variant="primary")        
+    submit_btn.click(commit, model_id, output)
 
 demo.launch()