Update pdf_server.py

magic_pdf/tools/pdf_server.py

@@ -65,8 +65,10 @@ def ocr_pdf_serve(args: str):
     os.environ["CUDA_VISIBLE_DEVICES"] = args.dcu_id
     config = configparser.ConfigParser()
     config.read(args.config_path)
-    host = config.get('server', 'pdf_host')
-    port = int(config.get('server', 'pdf_port'))
+    # host = config.get('server', 'pdf_host')
+    # port = int(config.get('server', 'pdf_port'))
+    pdf_server = config.get('server', 'pdf_server').split('://')[1]
+    host, port = pdf_server.split(':')[0], int(pdf_server.split(':')[1])
     global config_path
     config_path = args.config_path
     uvicorn.run(app, host=host, port=port)