Update app.py

app.py

@@ -7,20 +7,19 @@ import zipfile
 from pathlib import Path
 import re
 
-# os.system('pip install -r requirements.txt')
 # os.system('pip install -U magic-pdf==0.8.1')
 os.system('pip install -U magic_pdf-0.9.0a9-py3-none-any.whl')
-# os.system('python -m pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/')
 
-from huggingface_hub import snapshot_download
-model_dir = snapshot_download('opendatalab/PDF-Extract-Kit')
-layoutreader_model_dir = snapshot_download('hantian/layoutreader')
+# from huggingface_hub import snapshot_download
+# model_dir = snapshot_download('opendatalab/PDF-Extract-Kit')
+# layoutreader_model_dir = snapshot_download('hantian/layoutreader')
 
+# os.system('cp magic-pdf.template.json ~/magic-pdf.json')
+# os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /home/user/magic-pdf.json")
+# os.system(f"sed -i 's|/tmp/layoutreader|{layoutreader_model_dir}|g' /home/user/magic-pdf.json")
 
-# os.system('wget https://github.com/opendatalab/MinerU/raw/master/magic-pdf.template.json')
-os.system('cp magic-pdf.template.json ~/magic-pdf.json')
-os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /home/user/magic-pdf.json")
-os.system(f"sed -i 's|/tmp/layoutreader|{layoutreader_model_dir}|g' /home/user/magic-pdf.json")
+os.system('wget https://github.com/opendatalab/MinerU/raw/dev/docs/download_models_hf.py -O download_models_hf.py')
+os.system('python download_models_hf.py')
 os.system("sed -i 's|cpu|cuda|g' /home/user/magic-pdf.json")
 
 os.system('cp -r paddleocr /home/user/.paddleocr')