Update app.py

app.py

@@ -12,7 +12,8 @@ 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"
@@ -31,8 +32,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 == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
+    if doFaceEnhancer:
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
@@ -43,7 +44,7 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.video_encoder = "libx264"
     roop.globals.video_quality = 18
     roop.globals.max_memory = suggest_max_memory()
-    roop.globals.execution_providers = decode_execution_providers(["cpu"])
+    roop.globals.execution_providers = decode_execution_providers(["cuda"])
     roop.globals.execution_threads = suggest_execution_threads()
 
     print(
@@ -63,7 +64,20 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     return output_path
 
 
-app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="Apply", info="Face Enhancer")], outputs="image", theme="NoCrypt/miku", description="Sorry for the inconvenience. The model is currently running using CPU, which might affect performance. We appreciate your understanding."
-)
-app.launch()
+html_section_1 = "<div><h1>Welcome to the Face Swap App</h1></div>"
+html_section_2 = '<div><p>Upload your source and target images to swap faces. Optionally, use the face enhancer feature for HD Results.</p><h2><br /><strong>For fast and bulk Swap:</strong>&nbsp;<a href="https://picfy.xyz/" target="_blank" rel="noopener">https://picfy.xyz/</a><br /> <strong>Support me USDT (TRC-20): TAe7hsSVWtMEYz3G5V1UiUdYPQVqm28bKx</h2></div>'
+
+app = gr.Blocks()
+
+with app:
+    gr.HTML(html_section_1)
+    gr.HTML(html_section_2)
+    gr.Interface(
+        fn=swap_face,
+        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="Apply", info="Face Enhancer")],
+        outputs="image",
+        theme="NoCrypt/miku",
+        description="Sorry for the inconvenience. The model is currently running using CPU, which might affect performance. We appreciate your understanding."
+    )
+
+app.launch()