michael-qiu commited on
Commit
319bdd8
·
verified ·
1 Parent(s): 7df404e

Upload bailingmm_utils.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. bailingmm_utils.py +3 -3
bailingmm_utils.py CHANGED
@@ -16,10 +16,10 @@ from typing import Union, Tuple, List
16
 
17
  VIDEO_FETCH_VERSION = os.environ.get("VIDEO_FETCH_VERSION", "v1")
18
  if VIDEO_FETCH_VERSION == "v1":
19
- from bailingmm_utils_video import v1_fetch_video as fetch_video
20
  else:
21
- from bailingmm_utils_video import v2_fetch_video as fetch_video
22
- from bailingmm_utils_video import VideoInput
23
 
24
  logger = logging.getLogger(__name__)
25
 
 
16
 
17
  VIDEO_FETCH_VERSION = os.environ.get("VIDEO_FETCH_VERSION", "v1")
18
  if VIDEO_FETCH_VERSION == "v1":
19
+ from .bailingmm_utils_video import v1_fetch_video as fetch_video
20
  else:
21
+ from .bailingmm_utils_video import v2_fetch_video as fetch_video
22
+ from .bailingmm_utils_video import VideoInput
23
 
24
  logger = logging.getLogger(__name__)
25