formatting

app.py

@@ -1,17 +1,19 @@
-from huggingface_hub import list_datasets, list_models
-from cachetools import TTLCache, cached
+import os
 import platform
 import re
+from collections import defaultdict
+
 import gradio as gr
-from huggingface_hub import get_collection
+from cachetools import TTLCache, cached
 from cytoolz import groupby
-from collections import defaultdict
-import os
+from huggingface_hub import get_collection, list_datasets, list_models
 from tqdm.auto import tqdm
+
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 is_macos = platform.system() == "Darwin"
 LIMIT = None
-CACHE_TIME = 60 * 5 # 5 minutes
+CACHE_TIME = 60 * 5  # 5 minutes
+
 
 @cached(cache=TTLCache(maxsize=100, ttl=CACHE_TIME))
 def get_models():
@@ -113,5 +115,6 @@ def get_papers_for_collection(collection_slug: str):
 
 
 url = "HF-IA-archiving/models-to-archive-65006a7fdadb8c628f33aac9"
+slug_input = gr.Text(url, interactive=True, label="Collection slug", max_lines=1)
 
-gr.Interface(get_papers_for_collection, "text", "json").launch()
+gr.Interface(get_papers_for_collection, slug_input, "json").launch()