pass token around

app.py

@@ -37,7 +37,7 @@ BENCHMARKS_REPO_ID = "optimum-benchmark/OpenVINO-Benchmarks"
 TASKS = set(TASKS_TO_OVMODEL.keys()) & set(TASKS_TO_MODEL_LOADERS.keys())
 
 
-def parse_configs(inputs):
+def parse_configs(inputs, token: str):
     configs = {"process": {}, "inference": {}, "pytorch": {}, "openvino": {}}
 
     for key, value in inputs.items():
@@ -59,10 +59,18 @@ def parse_configs(inputs):
     configs["process"] = ProcessConfig(**configs.pop("process"))
     configs["inference"] = InferenceConfig(**configs.pop("inference"))
     configs["pytorch"] = PyTorchConfig(
-        task=task, model=model, device=DEVICE, **configs["pytorch"]
+        task=task,
+        model=model,
+        device=DEVICE,
+        **configs["pytorch"],
+        model_kwargs={"token": token},
     )
     configs["openvino"] = OVConfig(
-        task=task, model=model, device=DEVICE, **configs["openvino"]
+        task=task,
+        model=model,
+        device=DEVICE,
+        **configs["openvino"],
+        model_kwargs={"token": token},
     )
 
     return configs
@@ -79,8 +87,8 @@ def run_benchmark(inputs, oauth_token: Optional[gr.OAuthToken]):
     gr.Info(f"📩 Benchmark will be saved under {BENCHMARKS_REPO_ID}/{folder}")
 
     outputs = {backend: "Running..." for backend in BACKENDS}
+    configs = parse_configs(inputs, token=oauth_token.token)
     yield tuple(outputs[b] for b in BACKENDS)
-    configs = parse_configs(inputs)
 
     for backend in BACKENDS:
         try:
@@ -93,6 +101,9 @@ def run_benchmark(inputs, oauth_token: Optional[gr.OAuthToken]):
             )
             benchmark_report = Benchmark.launch(benchmark_config)
 
+            # make sure the token is never pushed to the hub
+            del benchmark_config.backend.model_kwargs["token"]
+
             benchmark_config.push_to_hub(
                 repo_id=BENCHMARKS_REPO_ID,
                 subfolder=benchmark_name,