配置更改

docker-compose.yml

@@ -21,6 +21,8 @@ services:
     container_name: lae_python
     ports:
       - "8003:8003"
+    volumes:
+      - .:/app
     stdin_open: true
     tty: true
     command: sh -c "cd src && python gradio_app.py"

requirements.txt

@@ -19,3 +19,5 @@ faiss-cpu==1.7.2 # https://github.com/facebookresearch/faiss/blob/main/INSTALL.m
 gradio==3.48.0
 qianfan==0.3.13.1
 modelscope==1.14.0
+wkhtmltopdf==0.2
+pdfkit==1.0.0
\ No newline at end of file

src/config/consts.py

@@ -66,4 +66,4 @@ NLP_BERT_PATH = 'C:\\Users\\15663\\AI\\models\\nlp_bert_document-segmentation_ch
 # =============================
 # ICON配置
 # =============================
-ICON_PATH = '/app/test/icon'
+ICON_PATH = '/app/src/icon'
\ No newline at end of file

src/gradio_app.py

@@ -228,7 +228,7 @@ def main():
         #     reset_text, [], input_text
         # )
 
-    demo.queue().launch(share=False, inbrowser=True, server_name='192.168.22.32', server_port=8888)
+    demo.queue().launch(share=False, inbrowser=True, server_name='0.0.0.0', server_port=GR_PORT)
 
 
 if __name__ == "__main__":