Added login button

app.py

@@ -298,52 +298,57 @@ with demo:
                                 datatype=EVAL_TYPES,
                                 row_count=5,
                             )
-            with gr.Row():
-                gr.Markdown("# ✉️✨ Submit your model here!", elem_classes="markdown-text")
+                        
+            if gr.OAuthToken:
+                with gr.Row():
+                    gr.Markdown("# ✉️✨ Submit your model here!", elem_classes="markdown-text")
 
-            with gr.Row():
-                with gr.Column():
-                    model_name_textbox = gr.Textbox(label="Model name")
-                    revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
-                    model_type = gr.Dropdown(
-                        choices=[t.to_str(" : ") for t in ModelType if t != ModelType.Unknown],
-                        label="Model type",
-                        multiselect=False,
-                        value=None,
-                        interactive=True,
-                    )
+                with gr.Row():
+                    with gr.Column():
+                        model_name_textbox = gr.Textbox(label="Model name")
+                        revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
+                        model_type = gr.Dropdown(
+                            choices=[t.to_str(" : ") for t in ModelType if t != ModelType.Unknown],
+                            label="Model type",
+                            multiselect=False,
+                            value=None,
+                            interactive=True,
+                        )
 
-                with gr.Column():
-                    precision = gr.Dropdown(
-                        choices=[i.value.name for i in Precision if i != Precision.Unknown],
-                        label="Precision",
-                        multiselect=False,
-                        value="float16" if DEVICE != "cpu" else "float32",
-                        interactive=True,
-                    )
-                    weight_type = gr.Dropdown(
-                        choices=[i.value.name for i in WeightType],
-                        label="Weights type",
-                        multiselect=False,
-                        value="Original",
-                        interactive=True,
-                    )
-                    base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
+                    with gr.Column():
+                        precision = gr.Dropdown(
+                            choices=[i.value.name for i in Precision if i != Precision.Unknown],
+                            label="Precision",
+                            multiselect=False,
+                            value="float16" if DEVICE != "cpu" else "float32",
+                            interactive=True,
+                        )
+                        weight_type = gr.Dropdown(
+                            choices=[i.value.name for i in WeightType],
+                            label="Weights type",
+                            multiselect=False,
+                            value="Original",
+                            interactive=True,
+                        )
+                        base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
 
-            submit_button = gr.Button("Submit Eval")
-            submission_result = gr.Markdown()
-            submit_button.click(
-                add_new_eval,
-                [
-                    model_name_textbox,
-                    base_model_name_textbox,
-                    revision_name_textbox,
-                    precision,
-                    weight_type,
-                    model_type,
-                ],
-                submission_result,
-            )
+                gr.Markdown("You must be logged in to submit models.")
+                gr.LoginButton(min_width=250)
+
+                submit_button = gr.Button("Submit Eval")
+                submission_result = gr.Markdown()
+                submit_button.click(
+                    add_new_eval,
+                    [
+                        model_name_textbox,
+                        base_model_name_textbox,
+                        revision_name_textbox,
+                        precision,
+                        weight_type,
+                        model_type,
+                    ],
+                    submission_result,
+                )
 
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=1800)

requirements.txt

@@ -2,7 +2,7 @@ APScheduler==3.10.1
 black==23.11.0
 click==8.1.3
 datasets==2.14.5
-gradio==4.4.0
+gradio[oauth]==4.29.0
 gradio_client==0.7.0
 huggingface-hub>=0.18.0
 matplotlib==3.7.1

src/submission/submit.py

@@ -1,6 +1,8 @@
 import json
 import os
 from datetime import datetime, timezone
+import gradio as gr
+from huggingface_hub import HfApi
 
 from src.display.formatting import styled_error, styled_message, styled_warning
 from src.envs import API, EVAL_REQUESTS_PATH, TOKEN, QUEUE_REPO
@@ -21,7 +23,11 @@ def add_new_eval(
     precision: str,
     weight_type: str,
     model_type: str,
+    oauth_token: gr.OAuthToken
 ):
+    if not oauth_token:
+        return styled_error("Please login in order to submit models")
+    
     global REQUESTED_MODELS
     global USERS_TO_SUBMISSION_DATES
     if not REQUESTED_MODELS:
@@ -72,6 +78,10 @@ def add_new_eval(
     if not modelcard_OK:
         return styled_error(error_msg)
 
+    print('Getting user info...')
+    api = HfApi(token=oauth_token.token)
+    user_info = api.whoami()
+
     # Seems good, creating the eval
     print("Adding new eval")
 
@@ -88,8 +98,9 @@ def add_new_eval(
         "params": model_size,
         "license": license,
         "private": False,
+        "user_info": user_info
     }
-
+    
     # Check for duplicate submission
     if f"{model}_{revision}_{precision}" in REQUESTED_MODELS:
         return styled_warning("This model has been already submitted.")