File tree Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Original file line number Diff line number Diff line change 1
- ## 0.7.37-dev0
1
+ ## 0.7.37-dev1
2
2
3
3
* refactor: remove layout analysis related code
4
+ * enhancement: Hide warning about table transformer weights not being loaded
4
5
5
6
## 0.7.36
6
7
Original file line number Diff line number Diff line change 1
- __version__ = "0.7.37-dev0 " # pragma: no cover
1
+ __version__ = "0.7.37-dev1 " # pragma: no cover
Original file line number Diff line number Diff line change 9
9
import numpy as np
10
10
import torch
11
11
from PIL import Image as PILImage
12
- from transformers import DetrImageProcessor , TableTransformerForObjectDetection
12
+ from transformers import DetrImageProcessor , TableTransformerForObjectDetection , logging
13
13
from transformers .models .table_transformer .modeling_table_transformer import (
14
14
TableTransformerObjectDetectionOutput ,
15
15
)
@@ -65,7 +65,10 @@ def initialize(
65
65
66
66
try :
67
67
logger .info ("Loading the table structure model ..." )
68
+ cached_current_verbosity = logging .get_verbosity ()
69
+ logging .set_verbosity_error ()
68
70
self .model = TableTransformerForObjectDetection .from_pretrained (model )
71
+ logging .set_verbosity (cached_current_verbosity )
69
72
self .model .eval ()
70
73
71
74
except EnvironmentError :
You can’t perform that action at this time.
0 commit comments