adding progress

src/submission/submit.py

@@ -1,7 +1,7 @@
 import json
 import os
 from datetime import datetime, timezone
-
+import gradio as gr
 from src.display.formatting import styled_error, styled_message, styled_warning
 from src.envs import API, EVAL_REQUESTS_PATH, TOKEN, QUEUE_REPO
 from src.submission.check_validity import (
@@ -17,6 +17,7 @@ USERS_TO_SUBMISSION_DATES = None
 
 def add_new_eval(
     model: str,
+    progress=gr.Progress()
     #base_model: str,
     #revision: str,
     #precision: str,
@@ -37,6 +38,9 @@ def add_new_eval(
     #precision = precision.split(" ")[0]
     current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
 
+
+    progress(0.1, desc=f"Checking model {model} on hub")
+
     if not is_model_on_hub(model_name=model, token=TOKEN, test_tokenizer=True): #revision=revision
         return styled_error("Model does not exist on HF Hub. Please select a valid model name.")
     
@@ -99,6 +103,7 @@ def add_new_eval(
         #"private": False,
     }
 
+    progress(0.5, desc=f"Checking previous submissions")
     # Check for duplicate submission
     if f"{model}" in REQUESTED_MODELS: #_{revision}_{precision}
         return styled_warning("This model has been already submitted.")
@@ -162,5 +167,5 @@ def add_new_eval(
 
 
     return styled_message(
-        "Your request has been submitted to the evaluation queue!\nPlease wait for up to an 15 minutes for the model to show in the PENDING list."
+        "Thank you for submitting your request! It has been placed in the evaluation queue. Please allow up to 15 minutes for the model to appear in the PENDING list"
     )