Bläddra i källkod

Create download_models.py

Xiaomeng Zhao 1 år sedan
förälder
incheckning
7ce807f43f
1 ändrade filer med 4 tillägg och 0 borttagningar
  1. 4 0
      docs/download_models.py

+ 4 - 0
docs/download_models.py

@@ -0,0 +1,4 @@
+# use modelscope sdk download models
+from modelscope import snapshot_download
+model_dir = snapshot_download('wanderkid/PDF-Extract-Kit')
+print(f"model dir is: {model_dir}/models")