Ver Fonte

chore: 更新表格结构预测模型版本

jingze_cheng há 6 meses atrás
pai
commit
be92760c62
2 ficheiros alterados com 2 adições e 1 exclusões
  1. 1 0
      Dockerfile
  2. 1 1
      server.py

+ 1 - 0
Dockerfile

@@ -100,6 +100,7 @@ stdout_logfile_maxbytes=50MB\n\
 " > /etc/supervisor/conf.d/be.conf
 RUN cd / && wget ftp://192.168.199.31/table-paddleclas.tgz && tar xvf table-paddleclas.tgz
 RUN cd / && wget ftp://192.168.199.31/table-paddleocr.tgz && tar xvf table-paddleocr.tgz
+RUN mkdir -p models/table && cd models/table && wget ftp://192.168.199.31/SLANet_ch/20240222/SLANet_ch.zip && unzip SLANet_ch.zip && mv SLANet_ch SLANet_latest && rm SLANet_ch.zip
 
 Add . /workspace
 EXPOSE 8080

+ 1 - 1
server.py

@@ -33,7 +33,7 @@ table_engine = CustomPPStructure(layout=False,
                            use_gpu=True,
                            show_log=True,
                            use_angle_cls=True,
-                           table_model_dir="models/table/SLANet_911")
+                           table_model_dir="models/table/SLANet_latest")
 
 cls_lock = threading.Lock()