jebin2 commited on
Commit
ff8d97c
·
1 Parent(s): 9981019
comic_panel_extractor/llm_panel_extractor.py CHANGED
@@ -85,7 +85,7 @@ class LLMPanelExtractor:
85
  image_width, image_height = input_image.size
86
 
87
  # Run YOLO detection
88
- detection_results = self.yolo_model.predict(source=input_image_path, device=common.get_device())
89
  first_detection_result = detection_results[0]
90
  newly_detected_boxes = None
91
  all_processed_boxes = []
 
85
  image_width, image_height = input_image.size
86
 
87
  # Run YOLO detection
88
+ detection_results = self.yolo_model.predict(source=input_image_path, device=common.get_device(), verbose=True)
89
  first_detection_result = detection_results[0]
90
  newly_detected_boxes = None
91
  all_processed_boxes = []