DawnC commited on
Commit
7007d91
·
verified ·
1 Parent(s): e6a18b7

Update image_processor.py

Browse files
Files changed (1) hide show
  1. image_processor.py +2 -1
image_processor.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import numpy as np
3
  import torch
4
  import cv2
@@ -24,7 +25,7 @@ class ImageProcessor:
24
  def __init__(self, use_llm=True, llm_model_path=None, enable_places365=True, places365_model_name='resnet50_places365'):
25
  """Initialize the image processor with required components"""
26
  print(f"Initializing ImageProcessor with use_llm={use_llm}, enable_places365={enable_places365}")
27
-
28
  try:
29
  # Initialize basic components first
30
  self.use_llm = use_llm
 
1
  import os
2
+ import logging
3
  import numpy as np
4
  import torch
5
  import cv2
 
25
  def __init__(self, use_llm=True, llm_model_path=None, enable_places365=True, places365_model_name='resnet50_places365'):
26
  """Initialize the image processor with required components"""
27
  print(f"Initializing ImageProcessor with use_llm={use_llm}, enable_places365={enable_places365}")
28
+ self.logger = logging.getLogger(__name__)
29
  try:
30
  # Initialize basic components first
31
  self.use_llm = use_llm