Spaces:
Runtime error
Runtime error
Commit
Β·
613c664
1
Parent(s):
9f363d7
fix it
Browse files- deoldify/visualize.py +4 -4
deoldify/visualize.py
CHANGED
|
@@ -424,7 +424,7 @@ def get_artistic_video_colorizer(
|
|
| 424 |
|
| 425 |
|
| 426 |
def get_stable_video_colorizer(
|
| 427 |
-
root_folder: Path = Path('
|
| 428 |
weights_name: str = 'ColorizeVideo_gen',
|
| 429 |
results_dir='result_images',
|
| 430 |
render_factor: int = 21
|
|
@@ -436,7 +436,7 @@ def get_stable_video_colorizer(
|
|
| 436 |
|
| 437 |
|
| 438 |
def get_image_colorizer(
|
| 439 |
-
root_folder: Path = Path('
|
| 440 |
) -> ModelImageVisualizer:
|
| 441 |
if artistic:
|
| 442 |
return get_artistic_image_colorizer(root_folder=root_folder, render_factor=render_factor)
|
|
@@ -445,7 +445,7 @@ def get_image_colorizer(
|
|
| 445 |
|
| 446 |
|
| 447 |
def get_stable_image_colorizer(
|
| 448 |
-
root_folder: Path = Path('
|
| 449 |
weights_name: str = 'ColorizeStable_gen',
|
| 450 |
results_dir='result_images',
|
| 451 |
render_factor: int = 35
|
|
@@ -457,7 +457,7 @@ def get_stable_image_colorizer(
|
|
| 457 |
|
| 458 |
|
| 459 |
def get_artistic_image_colorizer(
|
| 460 |
-
root_folder: Path = Path('
|
| 461 |
weights_name: str = 'ColorizeArtistic_gen',
|
| 462 |
results_dir='result_images',
|
| 463 |
render_factor: int = 35
|
|
|
|
| 424 |
|
| 425 |
|
| 426 |
def get_stable_video_colorizer(
|
| 427 |
+
root_folder: Path = Path('./'),
|
| 428 |
weights_name: str = 'ColorizeVideo_gen',
|
| 429 |
results_dir='result_images',
|
| 430 |
render_factor: int = 21
|
|
|
|
| 436 |
|
| 437 |
|
| 438 |
def get_image_colorizer(
|
| 439 |
+
root_folder: Path = Path('./'), render_factor: int = 35, artistic: bool = True
|
| 440 |
) -> ModelImageVisualizer:
|
| 441 |
if artistic:
|
| 442 |
return get_artistic_image_colorizer(root_folder=root_folder, render_factor=render_factor)
|
|
|
|
| 445 |
|
| 446 |
|
| 447 |
def get_stable_image_colorizer(
|
| 448 |
+
root_folder: Path = Path('./'),
|
| 449 |
weights_name: str = 'ColorizeStable_gen',
|
| 450 |
results_dir='result_images',
|
| 451 |
render_factor: int = 35
|
|
|
|
| 457 |
|
| 458 |
|
| 459 |
def get_artistic_image_colorizer(
|
| 460 |
+
root_folder: Path = Path('./'),
|
| 461 |
weights_name: str = 'ColorizeArtistic_gen',
|
| 462 |
results_dir='result_images',
|
| 463 |
render_factor: int = 35
|