Update app.py

app.py

@@ -9,16 +9,19 @@ 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.0a8-py3-none-any.whl')
+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')
+
 
 # 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("sed -i 's|cpu|cuda|g' /home/user/magic-pdf.json")
+os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /root/magic-pdf.json")
+os.system(f"sed -i 's|/tmp/layoutreader|{layoutreader_model_dir}|g' /root/magic-pdf.json")
+os.system("sed -i 's|cpu|cuda|g' /root/magic-pdf.json")
 
 os.system('cp -r paddleocr /home/user/.paddleocr')