Update model_build.py

model_build.py

@@ -21,7 +21,7 @@ from models.psp import pSp
 
 def build_stylegan2(
     increment = 0.01,
-    network_pkl = 'pretrained/furry.pkl',
+    network_pkl = 'pretrained/ohayou_face.pkl',
     process = 'image',                 #['image', 'interpolation','truncation','interpolation-truncation']
     random_seed = 0,
     diameter = 100.0,
@@ -54,7 +54,7 @@ def build_stylegan2(
 def build_psp():
     test_opts = easydict.EasyDict({
         # arguments for inference script
-        'checkpoint_path' : 'pretrained/psp.pt',
+        'checkpoint_path' : 'pretrained/ohayou_face.pt',
         'couple_outputs' : False,
         'resize_outputs' : False,