ali-ghamdan commited on
Commit
613c664
Β·
1 Parent(s): 9f363d7
Files changed (1) hide show
  1. 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('C:/Users/galquja/Desktop/Tools/tools/DeOldify'),
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('C:/Users/galquja/Desktop/Tools/tools/DeOldify'), 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,7 +445,7 @@ def get_image_colorizer(
445
 
446
 
447
  def get_stable_image_colorizer(
448
- root_folder: Path = Path('C:/Users/galquja/Desktop/Tools/tools/DeOldify'),
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('C:/Users/galquja/Desktop/Tools/tools/DeOldify'),
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