Downloader.

app.py

@@ -1,7 +1,46 @@
+import os
 import gradio as gr
+import datetime
+import tempfile
+from huggingface_hub import hf_hub_download
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def download_slow(repo_id):
+    os.environ.pop("HF_TRANSFER", None)
+    with tempfile.NamedDirectory() as workdir:
+        hf_hub_download(
+            repo_id,
+            filename="pytorch_model.bin",
+            force_download=True,
+            cache_dir=workdir,
+        )
+
+
+def download_fast(repo_id):
+    os.environ["HF_TRANSFER"] = "1"
+    with tempfile.NamedDirectory() as workdir:
+        hf_hub_download(
+            repo_id,
+            filename="pytorch_model.bin",
+            force_download=True,
+            cache_dir=workdir,
+        )
+
+
+def download(repo_id):
+    start = datetime.datetime.now()
+    download_slow(repo_id)
+    taken_slow = datetime.datetime.now() - start
+
+    start = datetime.datetime.now()
+    download_fast(repo_id)
+    taken_fast = datetime.datetime.now() - start
+
+    return f"""
+Slow : {taken_slow}
+Fast : {taken_fast}
+    """
+
+
+iface = gr.Interface(fn=download, inputs="text", outputs="text")
 iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+hf_transfer
+-e git+https://github.com/huggingface/huggingface_hub@hf_transfer#egg=huggingface_hub