jebin2 commited on
Commit
9843f6c
·
1 Parent(s): cf3d6df
Files changed (1) hide show
  1. comic_panel_extractor/config.py +1 -1
comic_panel_extractor/config.py CHANGED
@@ -12,7 +12,7 @@ class Config:
12
  current_path = os.path.abspath(os.path.join(os.path.dirname(__file__)))
13
  YOLO_BASE_MODEL_NAME = os.getenv('YOLO_BASE_MODEL_NAME', 'yolo11s-seg')
14
  yolo_base_model_path: str = f'{current_path}/{YOLO_BASE_MODEL_NAME}.pt'
15
- YOLO_MODEL_NAME = f'{os.getenv('YOLO_MODEL_NAME', 'comic_panel')}_{YOLO_BASE_MODEL_NAME}'
16
  yolo_trained_model_path: str = f'{current_path}/{YOLO_MODEL_NAME}.pt'
17
  black_overlay_input_path: str = ""
18
  output_folder: str = "temp_dir"
 
12
  current_path = os.path.abspath(os.path.join(os.path.dirname(__file__)))
13
  YOLO_BASE_MODEL_NAME = os.getenv('YOLO_BASE_MODEL_NAME', 'yolo11s-seg')
14
  yolo_base_model_path: str = f'{current_path}/{YOLO_BASE_MODEL_NAME}.pt'
15
+ YOLO_MODEL_NAME = f"{os.getenv('YOLO_MODEL_NAME', 'comic_panel')}_{YOLO_BASE_MODEL_NAME}"
16
  yolo_trained_model_path: str = f'{current_path}/{YOLO_MODEL_NAME}.pt'
17
  black_overlay_input_path: str = ""
18
  output_folder: str = "temp_dir"