Update app.py

app.py

@@ -20,7 +20,7 @@ API = HfApi(token=HF_TOKEN)
 @lru_cache(maxsize=128)
 def _list_collection_files(pr_number: str) -> Tuple[str, ...]:
     """
-    Return the `collection_summary.json` paths stored for a specific PR.
+    Return the `failure_summary.json` paths stored for a specific PR.
     """
     prefix = f"pr-{pr_number}"
     print(f"Looking for files with prefix: {prefix}")
@@ -29,6 +29,7 @@ def _list_collection_files(pr_number: str) -> Tuple[str, ...]:
         entries = API.list_repo_tree(
             repo_id=DATASET_ID,
             repo_type="dataset",
+            revision="main",
             recursive=True,
         )
     except HfHubHTTPError as error:
@@ -42,12 +43,12 @@ def _list_collection_files(pr_number: str) -> Tuple[str, ...]:
         # Debug: show all files that match the prefix
         if entry_type == "file" and entry.path.startswith(prefix):
             matching_paths.append(entry.path)
-        # Filter by prefix and look for collection_summary.json files
-        if entry_type == "file" and entry.path.startswith(prefix) and entry.path.endswith("collection_summary.json"):
+        # Filter by prefix and look for failure_summary.json files
+        if entry_type == "file" and entry.path.startswith(prefix) and entry.path.endswith("failure_summary.json"):
             files.append(entry.path)
     
     print(f"Found {len(matching_paths)} files with prefix {prefix}")
-    print(f"Found {len(files)} collection_summary.json files")
+    print(f"Found {len(files)} failure_summary.json files")
     if matching_paths:
         print(f"Sample matching paths: {matching_paths[:5]}")
     return tuple(files)
@@ -156,6 +157,8 @@ with gr.Blocks() as demo:
 
         Use the filters below to inspect CircleCI test aggregation records for the Transformers repository (or any
         repository that uploads data to the `transformers-community/circleci-test-results` dataset).
+        
+        Files are named `failure_summary.json` and organized as `pr-{PR}/sha-{COMMIT}/failure_summary.json`.
         """
     )