test

README.md

@@ -8,7 +8,9 @@ sdk_version: 5.8.0
 app_file: app.py
 hf_oauth: true
 hf_oauth_scopes:
-- read-repos
+  - read-repos
+  - write-repos
+  - manage-repos
 pinned: false
 license: apache-2.0
 ---

app.py

@@ -41,8 +41,6 @@ DEVICE = "cpu"
 LAUNCHER = "process"
 SCENARIO = "inference"
 BACKENDS = ["pytorch", "openvino"]
-BENCHMARKS_HF_TOKEN = os.getenv("BENCHMARKS_HF_TOKEN")
-BENCHMARKS_REPO_ID = "optimum-benchmark/OpenVINO-Benchmarks"
 TASKS = set(TASKS_TO_OVMODELS.keys() | TASKS_TO_OVPIPELINES) & set(
     TASKS_TO_AUTO_MODEL_CLASS_NAMES.keys() | TASKS_TO_AUTO_PIPELINE_CLASS_NAMES.keys()
 )
@@ -66,7 +64,12 @@ def parse_configs(inputs):
 
     for key in configs.keys():
         for k, v in configs[key].items():
-            if k in ["input_shapes", "generate_kwargs", "call_kwargs", "numactl_kwargs"]:
+            if k in [
+                "input_shapes",
+                "generate_kwargs",
+                "call_kwargs",
+                "numactl_kwargs",
+            ]:
                 configs[key][k] = eval(v)
 
     configs["process"] = ProcessConfig(**configs.pop("process"))
@@ -93,10 +96,11 @@ def run_benchmark(inputs, oauth_token: Optional[gr.OAuthToken]):
         raise gr.Error("Please login to be able to run the benchmark.")
 
     timestamp = time.strftime("%Y-%m-%d-%H-%M-%S")
-    use_name = whoami(oauth_token.token)["name"]
-    folder = f"{use_name}/{timestamp}"
+    user_name = whoami(oauth_token.token)["name"]
+    repo_id = f"{user_name}/benchmarks"
+    folder = f"{timestamp}"
 
-    gr.Info(f"📩 Benchmark will be saved under {BENCHMARKS_REPO_ID}/{folder}")
+    gr.Info(f"📩 Benchmark will be saved under {repo_id} in the folder {folder}")
 
     outputs = {backend: "Running..." for backend in BACKENDS}
     configs = parse_configs(inputs)
@@ -114,14 +118,14 @@ def run_benchmark(inputs, oauth_token: Optional[gr.OAuthToken]):
             benchmark_report = Benchmark.launch(benchmark_config)
 
             benchmark_config.push_to_hub(
-                repo_id=BENCHMARKS_REPO_ID,
+                repo_id=repo_id,
                 subfolder=benchmark_name,
-                token=BENCHMARKS_HF_TOKEN,
+                token=oauth_token.token,
             )
             benchmark_report.push_to_hub(
-                repo_id=BENCHMARKS_REPO_ID,
+                repo_id=repo_id,
                 subfolder=benchmark_name,
-                token=BENCHMARKS_HF_TOKEN,
+                token=oauth_token.token,
             )
 
         except GatedRepoError: