docker build 构建修改

2025-11-06 17:15:46 +08:00
parent 33e09afc54
commit ecdb4321e8
58 changed files with 72216 additions and 237 deletions
--- a/rag/res/deepdoc/README.md
+++ b/rag/res/deepdoc/README.md
@@ -0,0 +1,19 @@
+---
+license: apache-2.0
+---
+
+### Model Loading
+```python
+import xgboost as xgb
+import torch
+
+model = xgb.Booster()
+if torch.cuda.is_available():
+model.set_param({"device": "cuda"})
+model.load_model('InfiniFlow/text_concat_xgb_v1.0')
+```
+
+### Prediction
+```python
+model.predict(xgb.DMatrix([feature]))[0]
+```
--- a/rag/res/deepdoc/det.onnx
+++ b/rag/res/deepdoc/det.onnx
--- a/rag/res/deepdoc/ocr.res
+++ b/rag/res/deepdoc/ocr.res
--- a/rag/res/deepdoc/rec.onnx
+++ b/rag/res/deepdoc/rec.onnx
--- a/rag/res/deepdoc/updown_concat_xgb.model
+++ b/rag/res/deepdoc/updown_concat_xgb.model