Update app.py

app.py

@@ -23,14 +23,14 @@ if not os.path.exists('CodeFormer.pth'):
     os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/CodeFormer.pth -P .")
 
 torch.hub.download_url_to_file(
-    'https://upload.wikimedia.org/wikipedia/commons/thumb/a/ab/Abraham_Lincoln_O-77_matte_collodion_print.jpg/1024px-Abraham_Lincoln_O-77_matte_collodion_print.jpg',
-    'lincoln.jpg')
+    'https://haoluobo.com/wp-content/uploads/2023/01/%E6%A8%B1%E6%9C%A8%E8%8A%B1%E9%81%93.jpg',
+    'HanamichiSakuragi.jpg')
 torch.hub.download_url_to_file(
-    'https://user-images.githubusercontent.com/17445847/187400315-87a90ac9-d231-45d6-b377-38702bd1838f.jpg',
-    'AI-generate.jpg')
+    'https://haoluobo.com/wp-content/uploads/2023/01/%E6%9D%8E%E4%B8%96%E6%B0%91.jpg',
+    'LiShiming.jpg')
 torch.hub.download_url_to_file(
-    'https://user-images.githubusercontent.com/17445847/187400981-8a58f7a4-ef61-42d9-af80-bc6234cef860.jpg',
-    'Blake_Lively.jpg')
+    'https://haoluobo.com/wp-content/uploads/2023/01/%E4%B9%BE%E9%9A%86.jpg',
+    'QianLong.jpg')
 torch.hub.download_url_to_file(
     'https://user-images.githubusercontent.com/17445847/187401133-8a3bf269-5b4d-4432-b2f0-6d26ee1d3307.png',
     '10045.png')
@@ -45,7 +45,12 @@ os.makedirs('output', exist_ok=True)
 
 
 # def inference(img, version, scale, weight):
-def inference(img, version, scale):
+def inference(img, version, scale, blur_face):
+    blur_face = int(blur_face)
+    if blur_face < 3:
+        blur_face = 3
+    if blur_face % 2 != 1:
+        blur_face += 1
     # weight /= 100
     print(img, version, scale)
     if scale > 4:
@@ -83,6 +88,12 @@ def inference(img, version, scale):
 
         try:
             # _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True, weight=weight)
+            face_helper = face_enhancer.face_helper
+            align_warp_face = face_helper.align_warp_face
+            def new_align_warp_face(*args, **kwargs):
+                align_warp_face(*args, **kwargs)  # save_cropped_path
+                # face_helper.cropped_faces = [cv2.GaussianBlur(e, (blur_face, blur_face), 0) for e in face_helper.cropped_faces]
+            face_helper.align_warp_face = new_align_warp_face
             _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
         except RuntimeError as error:
             print('Error', error)
@@ -131,6 +142,7 @@ demo = gr.Interface(
         # gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", default='v1.4', label='version'),
         gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], type="value", default='v1.4', label='version'),
         gr.inputs.Number(label="Rescaling factor", default=2),
+        gr.inputs.Number(label="Blur face", default=25),
         # gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
     ], [
         gr.outputs.Image(type="numpy", label="Output (The whole image)"),