revert to working light

app.py

@@ -2,7 +2,6 @@ import spaces
 import gradio as gr
 import cv2
 import traceback
-
 @spaces.GPU
 def swap_face(source_img, target_img):
     import insightface
@@ -22,39 +21,29 @@ def swap_face(source_img, target_img):
         if len(s_faces)==0 or len(t_faces)==0:
             app2 = FaceAnalysis(name='buffalo_l')
             app2.prepare(ctx_id=-1, det_size=(1024,1024))
-            if len(s_faces)==0:
-                s_faces = app2.get(src)
-            if len(t_faces)==0:
-                t_faces = app2.get(tgt)
+            if len(s_faces)==0: s_faces = app2.get(src)
+            if len(t_faces)==0: t_faces = app2.get(tgt)
         if len(s_faces)==0 or len(t_faces)==0:
             cv2.putText(tgt, f"s:{len(s_faces)} t:{len(t_faces)}", (20,40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,255),2)
             out="/tmp/debug.jpg"
             cv2.imwrite(out,tgt)
             return out
         res = swapper.get(tgt, t_faces[0], s_faces[0], paste_back=True)
-        # High quality GFPGAN enhancer
-        try:
-            from gfpgan import GFPGANer
-            enhancer = GFPGANer(model_path='https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth', upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
-            _, _, res = enhancer.enhance(res, has_aligned=False, only_center_face=False, paste_back=True)
-        except Exception as e:
-            print("gfpgan fail",e, flush=True)
-        out="/tmp/swapped_gfpgan.jpg"
+        out="/tmp/swapped.jpg"
         cv2.imwrite(out,res)
         return out
     except Exception as e:
         tgt2=cv2.imread(target_img)
-        cv2.putText(tgt2, str(e)[:60], (20,40), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0,0,255),2)
+        cv2.putText(tgt2, str(e)[:50], (20,40), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0,0,255),2)
         out="/tmp/err.jpg"
         cv2.imwrite(out,tgt2)
         return out
-
 with gr.Blocks() as demo:
-    gr.Markdown("# FaceFusion Zero + GFPGAN HQ")
+    gr.Markdown("# FaceFusion Zero")
     with gr.Row():
         src = gr.Image(type="filepath", label="Source")
         tgt = gr.Image(type="filepath", label="Target")
-    out = gr.Image(type="filepath", label="GFPGAN Enhanced")
-    btn = gr.Button("Swap + GFPGAN")
+    out = gr.Image(type="filepath", label="Result")
+    btn = gr.Button("Swap")
     btn.click(swap_face, inputs=[src, tgt], outputs=out)
 demo.launch()

requirements.txt

@@ -3,9 +3,5 @@ spaces
 insightface==0.7.3
 onnxruntime-gpu
 opencv-python==4.8.0.76
-gfpgan==1.3.8
-basicsr==1.4.2
-facexlib==0.3.0
-realesrgan==0.3.0
 huggingface_hub
 hf_transfer