Hanzo03 commited on
Commit
4bc8241
·
verified ·
1 Parent(s): 5172b77

Update utils/engine.py

Browse files
Files changed (1) hide show
  1. utils/engine.py +2 -2
utils/engine.py CHANGED
@@ -7,8 +7,8 @@ import zarr
7
  from PIL import Image
8
  from typing import Tuple, List
9
 
10
- from config import config, get_logger
11
- from models import device, clip_processor, clip_model, collection, chroma_client, vlm_model, vlm_tokenizer
12
  logger = get_logger("Engine")
13
 
14
  def process_and_index_video(video_path: str) -> Tuple[str, List[Image.Image]]:
 
7
  from PIL import Image
8
  from typing import Tuple, List
9
 
10
+ from utils.config import config, get_logger
11
+ from utils.models import device, clip_processor, clip_model, collection, chroma_client, vlm_model, vlm_tokenizer
12
  logger = get_logger("Engine")
13
 
14
  def process_and_index_video(video_path: str) -> Tuple[str, List[Image.Image]]: