fix

app.py

@@ -1,6 +1,8 @@
 import os
 import time
 import traceback
+import logging
+from typing import Optional
 
 from config_store import (
     get_process_config,
@@ -44,14 +46,16 @@ MODELS_TO_TASKS = {
 TASKS = set(TASKS_TO_OVMODEL.keys()) & set(TASKS_TO_MODEL_LOADERS.keys())
 
 
-def run_benchmark(kwargs, oauth_token: gr.OAuthToken):
-    if oauth_token.token is None:
+def run_benchmark(kwargs, oauth_token: Optional[gr.OAuthToken] = None):
+    if oauth_token.token is None or oauth_token.token == "":
         gr.Error("Please login to be able to run the benchmark.")
         return tuple(None for _ in BACKENDS)
 
+    setup_logging(level="INFO", prefix="MAIN-PROCESS")
+
     timestamp = time.strftime("%Y-%m-%d-%H-%M-%S")
-    username = whoami(oauth_token.token)["name"]
-    repo_id = f"{username}/benchmarks"
+    name = whoami(oauth_token.token)["name"]
+    repo_id = f"{name}/benchmarks"
     token = oauth_token.token
 
     create_repo(repo_id, token=token, repo_type="dataset", exist_ok=True)
@@ -124,7 +128,7 @@ def run_benchmark(kwargs, oauth_token: gr.OAuthToken):
             benchmark.push_to_hub(
                 repo_id=repo_id, subfolder=benchmark_name, token=oauth_token.token
             )
-            gr.Info(f"Pushed benchmark to {username}/benchmarks/{benchmark_name}")
+            gr.Info(f"Pushed benchmark to {repo_id}/{benchmark_name}")
 
             outputs[backend] = f"\n{benchmark_report.to_markdown_text()}"
 
@@ -137,11 +141,13 @@ def run_benchmark(kwargs, oauth_token: gr.OAuthToken):
 
             yield tuple(outputs[b] for b in BACKENDS)
 
+    logging.getLogger().setLevel(logging.NOTSET)
+
 
 def build_demo():
     with gr.Blocks() as demo:
         # add login button
-        gr.LoginButton(min_width=250)
+        gr.LoginButton()
 
         # add image
         gr.HTML(
@@ -237,11 +243,8 @@ def build_demo():
     return demo
 
 
-demo = build_demo()
-
 if __name__ == "__main__":
     os.environ["LOG_TO_FILE"] = "0"
     os.environ["LOG_LEVEL"] = "INFO"
-    setup_logging(level="INFO", prefix="MAIN-PROCESS")
 
-    demo.queue(max_size=10).launch()
+    build_demo().queue(max_size=10).launch()