ZJU-AI4H commited on
Commit
c734c38
·
verified ·
1 Parent(s): 4eb9c5a

Update image_processing_hulumed.py

Browse files
Files changed (1) hide show
  1. image_processing_hulumed.py +5 -1
image_processing_hulumed.py CHANGED
@@ -39,7 +39,6 @@ from transformers.image_utils import (
39
  ChannelDimension,
40
  ImageInput,
41
  PILImageResampling,
42
- VideoInput,
43
  get_image_size,
44
  infer_channel_dimension_format,
45
  is_scaled_image,
@@ -47,6 +46,11 @@ from transformers.image_utils import (
47
  make_list_of_images,
48
  to_numpy_array,
49
  )
 
 
 
 
 
50
  from transformers.utils import TensorType, is_vision_available, logging
51
 
52
 
 
39
  ChannelDimension,
40
  ImageInput,
41
  PILImageResampling,
 
42
  get_image_size,
43
  infer_channel_dimension_format,
44
  is_scaled_image,
 
46
  make_list_of_images,
47
  to_numpy_array,
48
  )
49
+ try:
50
+ from transformers.video_utils import VideoInput
51
+ except:
52
+ from transformers.image_utils import VideoInput
53
+
54
  from transformers.utils import TensorType, is_vision_available, logging
55
 
56