Merge branch 'main' of https://huggingface.co/spaces/Reeve/Ohayou_Face

app.py

@@ -19,13 +19,13 @@ pretransform = transforms.Compose([
 def pipeline(img):
     img = model_build.img_preprocess(img, pretransform)
     with torch.no_grad():
-        _, latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)
+        latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)
     img = stylegan2(latent_space, noise_mode='none')
     img = Image.fromarray(np.array((img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8).squeeze(0)[20:-20,:,:]))
     img.save('output.png')
     return 'output.png'
     
-examples=[['momoi_out.png',False], ['churuki_out.png', False], ['fgfgfggf.png', False], ['dsfd.png', False]]
-description="The male image doesn't work well. 1:1 ratio image recommended (square cropable after uploading). If the background is not monochromatic, it can be mixed with hair color. It takes an average of 5 seconds, but it can take longer if there is a lot of traffic. 남성 이미지에는 잘 작동하지 않음. 1:1비율 권장(업로드 후 정사각형 자르기 가능), 배경이 단색이 아니면 머리색과 섞일 수 있음. 트래픽이 많으면 5초 이상 걸릴 수 있음. Email:krkmfn@gmail.com"
+examples=[['1.png',False], ['2.png', False], ['3.png', False], ['4.png', False], ['5.png', False]]
+description="이미지 업로드 후 연필모양 아이콘 누르면 정사각형으로 자르기 할 수 있습니다. 디테일은 좀 떨어짐. 얼굴 각도(?)에 따라 잘 안될수도 있음. After uploading the image, you can square crop."
 gr.Interface(pipeline, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="file"),description=description,allow_flagging=False,examples=examples,allow_screenshot=False,enable_queue=False).launch()
     

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,