Add logging for download_models.py

classtranscribe · Feb 15, 2025 · 8d5f5f9 · 8d5f5f9
1 parent faff6e4
commit 8d5f5f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/download_models.py b/server/download_models.py
@@ -6,6 +6,7 @@
 # ///
 from huggingface_hub import snapshot_download
 
+print("Downloading models...")
 snapshot_download(
     repo_id="opendatalab/pdf-extract-kit-1.0",
     local_dir="./",
@@ -16,3 +17,4 @@
         "models/TabRec/*",
     ],
 )
+print("Download complete.")