format links

app.py

@@ -6,7 +6,9 @@ import gradio as gr
 from huggingface_hub import get_collection
 from cytoolz import groupby
 from collections import defaultdict
+import os
 
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 is_macos = platform.system() == "Darwin"
 LIMIT = None
 
@@ -20,8 +22,10 @@ def get_models():
 def get_datasets():
     return list(iter(list_datasets(full=True, limit=LIMIT)))
 
-get_models() # warm up the cache
-get_datasets() # warm up the cache
+
+get_models()  # warm up the cache
+get_datasets()  # warm up the cache
+
 
 def check_for_arxiv_id(model):
     return [tag for tag in model.tags if "arxiv" in tag] if model.tags else False
@@ -94,11 +98,14 @@ def get_papers_for_collection(collection_slug: str):
                 dataset_papers.append(arxiv_id)
     if collection_models is not None:
         for model in collection.get("models", []):
-            if arxiv_id := models_to_arxiv_id.get(model.item_id, None):
+            if arxiv_ids := models_to_arxiv_id.get(model.item_id, None):
                 data = {
                     "model": model.item_id,
-                    "arxiv_id": arxiv_id,
-                    "hub_paper_link": f"https://huggingface.co/papers/{arxiv_id}",
+                    "arxiv_ids": arxiv_ids,
+                    "hub_paper_links": [
+                        f"https://huggingface.co/papers/{arxiv_id}"
+                        for arxiv_id in arxiv_ids
+                    ],
                 }
                 model_papers.append(data)
     return {"datasets": dataset_papers, "models": model_papers}
@@ -106,4 +113,4 @@ def get_papers_for_collection(collection_slug: str):
 
 url = "HF-IA-archiving/models-to-archive-65006a7fdadb8c628f33aac9"
 
-gr.Interface(get_papers_for_collection,"text", "json").launch()
+gr.Interface(get_papers_for_collection, "text", "json").launch()