Spaces:
Running
Running
issue fix
Browse files
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
|
| 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"
|