Joker1212 commited on
Commit
405bf22
·
1 Parent(s): 86e1666

feat: add local config

Browse files

(cherry picked from commit 9ef94c4afb720afaa07ca745420859be147d587c)

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from rapid_undistorted.inference import InferenceEngine
3
  from rapid_unwrap.inference import DocUnwrapper
4
  import numpy as np
5
  # 初始化模型
6
- engine = InferenceEngine()
7
  # 添加示例
8
  example_images = [
9
  "images/demo.jpg",
 
3
  from rapid_unwrap.inference import DocUnwrapper
4
  import numpy as np
5
  # 初始化模型
6
+ engine = InferenceEngine("config.yaml")
7
  # 添加示例
8
  example_images = [
9
  "images/demo.jpg",