rodrigomasini commited on
Commit
cf28c06
·
verified ·
1 Parent(s): 8109ff4

Update mdr_pdf_parser.py

Browse files
Files changed (1) hide show
  1. mdr_pdf_parser.py +2 -2
mdr_pdf_parser.py CHANGED
@@ -31,7 +31,7 @@ import torch # PyTorch
31
  import requests # For downloading models
32
  from pathlib import Path
33
  from enum import auto, Enum
34
- from dataclasses import dataclass
35
  from typing import Literal, Iterable, Generator, Sequence, Callable, TypeAlias, List, Dict, Any, Optional
36
  from collections import defaultdict
37
  from math import pi, ceil, sin, cos, sqrt, atan2
@@ -1376,7 +1376,7 @@ class _MDR_ONNXParams:
1376
  cls_image_shape: str = "3,48,192"
1377
  cls_batch_num: int = 6
1378
  cls_thresh: float = 0.9
1379
- label_list: list = ['0', '180']
1380
 
1381
  # Attributes with default values (Group 2 - Detection)
1382
  det_algorithm: str = "DB"
 
31
  import requests # For downloading models
32
  from pathlib import Path
33
  from enum import auto, Enum
34
+ from dataclasses import dataclass, field
35
  from typing import Literal, Iterable, Generator, Sequence, Callable, TypeAlias, List, Dict, Any, Optional
36
  from collections import defaultdict
37
  from math import pi, ceil, sin, cos, sqrt, atan2
 
1376
  cls_image_shape: str = "3,48,192"
1377
  cls_batch_num: int = 6
1378
  cls_thresh: float = 0.9
1379
+ label_list: List[str] = field(default_factory=lambda: ['0', '180'])
1380
 
1381
  # Attributes with default values (Group 2 - Detection)
1382
  det_algorithm: str = "DB"