Update app.py

app.py

@@ -1,5 +1,5 @@
-# -* coding:UTF-8 -*
-# !/usr/bin/env python
+# -*- coding: UTF-8 -*-
+#!/usr/bin/env python
 import numpy as np
 import gradio as gr
 import roop.globals
@@ -11,22 +11,21 @@ from roop.core import (
 )
 from roop.processors.frame.core import get_frame_processors_modules
 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"
 
+    # Salva le immagini
     source_image = Image.fromarray(source_file)
     source_image.save(source_path)
     target_image = Image.fromarray(target_file)
     target_image.save(target_path)
 
-    print("source_path: ", source_path)
-    print("target_path: ", target_path)
+    print("source_path:", source_path)
+    print("target_path:", target_path)
 
     roop.globals.source_path = source_path
     roop.globals.target_path = target_path
@@ -34,10 +33,14 @@ 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"]
+
+    # Processori
+    if doFaceEnhancer:
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
+
+    # Impostazioni globali
     roop.globals.headless = True
     roop.globals.keep_fps = True
     roop.globals.keep_audio = True
@@ -46,7 +49,8 @@ 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(["cuda"])
+    # FORZA CPU
+    roop.globals.execution_providers = decode_execution_providers(["cpu"])
     roop.globals.execution_threads = suggest_execution_threads()
 
     print(
@@ -56,17 +60,25 @@ def swap_face(source_file, target_file,doFaceEnhancer):
         roop.globals.output_path,
     )
 
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
+    # Controllo pre-processori
+    for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
         if not frame_processor.pre_check():
             return
 
+    # Avvio roop
     start()
     return output_path
 
 
+# Interfaccia Gradio
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+    fn=swap_face,
+    inputs=[
+        gr.Image(label="Source Image"),
+        gr.Image(label="Target Image"),
+        gr.Checkbox(label="Face Enhancer?", info="Enable face enhancer?")
+    ],
+    outputs="image"
 )
+
 app.launch()