Update app.py

app.py

@@ -16,10 +16,19 @@ os.system('pip install git+https://github.com/opendatalab/MinerU.git@dev')
 
 os.system('wget https://github.com/opendatalab/MinerU/raw/master/scripts/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")
+
+with open('/home/user/magic-pdf.json', 'r') as file:
+    data = json.load(file)
+
+data['device-mode'] = "cuda"
+if os.getenv('apikey'):
+    data['llm-aided-config']['title_aided']['api_key'] = os.getenv('apikey')
+    data['llm-aided-config']['title_aided']['enable'] = True
+
+with open('/home/user/magic-pdf.json', 'w') as file:
+    json.dump(data, file, indent=4)
 
 os.system('cp -r paddleocr /home/user/.paddleocr')
-os.system('pip install -U gradio-pdf')
 from gradio_pdf import PDF
 
 import gradio as gr