Update app.py

app.py

@@ -1,3 +1,6 @@
+
+#### APP.PY CODE END ###
+
 import os
 import spaces
 import cv2
@@ -28,9 +31,9 @@ from utils import trim_video, StreamerThread, ProcessBar, open_directory, split_
 
 ## ------------------------------ USER ARGS ------------------------------
 
-parser = argparse.ArgumentParser(description="Swap Face Swapper")
+parser = argparse.ArgumentParser(description="Swap-Mukham Face Swapper")
 parser.add_argument("--out_dir", help="Default Output directory", default=os.getcwd())
-parser.add_argument("--batch_size", help="Gpu batch size", default=1)
+parser.add_argument("--batch_size", help="Gpu batch size", default=32)
 parser.add_argument("--cuda", action="store_true", help="Enable cuda", default=False)
 parser.add_argument(
     "--colab", action="store_true", help="Enable colab mode", default=False
@@ -76,8 +79,6 @@ FACE_ENHANCER_LIST = ["NONE"]
 FACE_ENHANCER_LIST.extend(get_available_enhancer_names())
 FACE_ENHANCER_LIST.extend(cv2_interpolations)
 
-## ------------------------------ SET EXECUTION PROVIDER ------------------------------
-
 ## ------------------------------ SET EXECUTION PROVIDER ------------------------------
 # Note: Non CUDA users may change settings here
 
@@ -88,21 +89,16 @@ if USE_CUDA:
     if "CUDAExecutionProvider" in available_providers:
         print("\n********** Running on CUDA **********\n")
         PROVIDER = ["CUDAExecutionProvider", "CPUExecutionProvider"]
-        cv2.setNumThreads(32)
     else:
         USE_CUDA = False
         print("\n********** CUDA unavailable running on CPU **********\n")
-        cv2.setNumThreads(1)
 else:
     USE_CUDA = False
     print("\n********** Running on CPU **********\n")
-    cv2.setNumThreads(1)
+
 device = "cuda" if USE_CUDA else "cpu"
 EMPTY_CACHE = lambda: torch.cuda.empty_cache() if device == "cuda" else None
 
-## ------------------------------ LOAD MODELS ------------------------------
-
-
 ## ------------------------------ LOAD MODELS ------------------------------
 
 def load_face_analyser_model(name="buffalo_l"):
@@ -138,7 +134,7 @@ load_face_swapper_model()
 ## ------------------------------ MAIN PROCESS ------------------------------
 
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=300, enable_queue=True)
 def process(
     input_type,
     image_path,
@@ -169,22 +165,9 @@ def process(
     global WORKSPACE
     global OUTPUT_FILE
     global PREVIEW
-    global USE_CUDA  # Access global variables
-    global device
-    global PROVIDER
-    global FACE_ANALYSER, FACE_SWAPPER, FACE_ENHANCER, FACE_PARSER, NSFW_DETECTOR
-
-    # Set CUDA usage and device
-    USE_CUDA = True
-    device = "cuda"
-    PROVIDER = ["CUDAExecutionProvider", "CPUExecutionProvider"]
-
-    # Reset models to None to reload them with GPU
-    FACE_ANALYSER = None
-    FACE_SWAPPER = None
-    FACE_ENHANCER = None
-    FACE_PARSER = None
-    NSFW_DETECTOR = None    ## ------------------------------ GUI UPDATE FUNC ------------------------------
+    WORKSPACE, OUTPUT_FILE, PREVIEW = None, None, None
+
+    ## ------------------------------ GUI UPDATE FUNC ------------------------------
 
     def ui_before():
         return (
@@ -944,3 +927,6 @@ if __name__ == "__main__":
         print("Running in colab mode")
 
     interface.launch()
+
+
+#### APP.PY CODE END ###