Update app.py

app.py

@@ -1,4 +1,4 @@
-# -*- coding:UTF-8 -*-
+# -* coding:UTF-8 -*
 # !/usr/bin/env python
 import numpy as np
 import gradio as gr
@@ -14,7 +14,9 @@ from roop.utilities import normalize_output_path
 import os
 from PIL import Image
 
-def swap_face(source_file, target_file, doFaceEnhancer):
+
+def swap_face(source_file, target_file,doFaceEnhancer):
+
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -32,8 +34,8 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
-    if doFaceEnhancer:
-        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
+    if doFaceEnhancer == True:
+        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
@@ -63,13 +65,14 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     start()
     return output_path
 
-# HTML code for the ad banners
+# HTML code for the ad banner
 ad_code = """
 <div style="text-align: center;">
 <iframe src="https://www.fiverr.com/gig_widgets?id=U2FsdGVkX1+9vQaosK0mq7xGXzIb/5hdFYIQsmHlpeMwFm5gingveDZKXBz2oJwV7ZwvToxseOx/KzKtr77ryB3qTb83/+GvCyI8OKpZcSgM0sDHzEPlZnqNeu8Db4y/IXlAceA1UcYT5Z3cW5vBLrgJ/WOpyMWSiyfwwMmGngjltQBexcivW7ukRd+/0yos7GCbfGjGdnLPHm2LB/CB9w2mA1i8cVv9LU+UUdj/O5KLRXtnychl9wAMkPTpmwMiyuAfUZy0nbK/Xa5O2UecmCvn7wicTr0TCTpLmIxaEh3YftXOVr9e36OGVYTmy5nlsgujkqJPI7wL0dXfon0ru4kvoycC7UD6m/whMoxrWZOf386qV2eEeXzk3vKO/emXUUfSavtofzSBtjLcyAOVhnHJg27PvVPgvFTQdy0o3F1M0DaHYTVW3Ln45MPYvMSH&affiliate_id=36184&strip_google_tagmanager=true" loading="lazy" data-with-title="false" class="fiverr_nga_frame" frameborder="0" height="350" width="100%" referrerpolicy="no-referrer-when-downgrade" data-mode="random_gigs" onload=" var frame = this; var script = document.createElement('script'); script.addEventListener('load', function() { window.FW_SDK.register(frame); }); script.setAttribute('src', 'https://www.fiverr.com/gig_widgets/sdk'); document.body.appendChild(script); " ></iframe>
 </div>
-"""
+	"""
 
+# HTML code for the ad banner
 ad_code2 = """
 <div style="text-align: center;">
     <a href="https://beta.publishers.adsterra.com/referral/UNXJYTziBP" target="_blank" style="display: inline-block;">
@@ -79,22 +82,10 @@ ad_code2 = """
         <img fetchpriority="high" decoding="async" width="468" height="120" src="https://ziverr.xyz/wp-content/uploads/2024/06/PASSIVE-1.gif" class="attachment-large size-large wp-image-1266" alt="">
     </a>
 </div>
-"""
-
-with gr.Blocks() as app:
-    gr.HTML(ad_code)
-    input_image1 = gr.Image()
-    input_image2 = gr.Image()
-    face_enhancer = gr.Checkbox(label="face_enhancer?", info="do face enhancer?")
-    output_image = gr.Image()
-    swap_face_button = gr.Button("Swap Face")
+	"""
 
-    swap_face_button.click(
-        fn=swap_face,
-        inputs=[input_image1, input_image2, face_enhancer],
-        outputs=output_image
-    )
-
-    gr.HTML(ad_code2)
 
+app = gr.Interface(
+    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],gr.HTML(ad_code2), outputs="image"
+)
 app.launch()