gahanmakwana commited on
Commit
d0f81fb
·
1 Parent(s): 30de1f3

change app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -13,7 +13,19 @@ app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
13
  os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
14
 
15
  # Initialize PaddleOCR once (CPU mode, English; angle_cls=False speeds up simple text)
16
- ocr = PaddleOCR(use_angle_cls=False, use_gpu=False, lang='en')
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
  @app.route('/', methods=['GET', 'POST'])
19
  def index():
 
13
  os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
14
 
15
  # Initialize PaddleOCR once (CPU mode, English; angle_cls=False speeds up simple text)
16
+
17
+
18
+ # create /tmp/ocr_models directory if it doesn't exist
19
+ os.makedirs('/tmp/ocr_models', exist_ok=True)
20
+
21
+ ocr = PaddleOCR(
22
+ use_angle_cls=False,
23
+ use_gpu=False,
24
+ lang='en',
25
+ det_model_dir='/tmp/ocr_models/det',
26
+ rec_model_dir='/tmp/ocr_models/rec',
27
+ cls_model_dir='/tmp/ocr_models/cls'
28
+ )
29
 
30
  @app.route('/', methods=['GET', 'POST'])
31
  def index():