Update app.py

app.py

@@ -4,9 +4,23 @@ import datetime
 import tempfile
 from huggingface_hub import hf_hub_download
 
+def download_very_slow(repo_id):
+    os.environ.pop("HF_TRANSFER", None)
+    os.environ["HF_CHUNK_SIZE"] = "1024"
+
+    with tempfile.TemporaryDirectory() as workdir:
+        hf_hub_download(
+            repo_id,
+            filename="pytorch_model.bin",
+            force_download=True,
+            cache_dir=workdir,
+        )
+
 
 def download_slow(repo_id):
     os.environ.pop("HF_TRANSFER", None)
+    os.environ["HF_CHUNK_SIZE"] = "10485760"
+
     with tempfile.TemporaryDirectory() as workdir:
         hf_hub_download(
             repo_id,
@@ -28,6 +42,10 @@ def download_fast(repo_id):
 
 
 def download(repo_id):
+    start = datetime.datetime.now()
+    download_very_slow(repo_id)
+    taken_slow = datetime.datetime.now() - start
+    
     start = datetime.datetime.now()
     download_slow(repo_id)
     taken_slow = datetime.datetime.now() - start