Update app.py

app.py

@@ -1,8 +1,11 @@
+import os
 import gradio as gr
 from gradio_client import Client
 
+hf_token = os.environ.get('HF_TOKEN')
+
 sdxl_client = Client("https://fffiloni-sdxl-dpo.hf.space/")
-faceswap_client = Client("https://tonyassi-face-swap.hf.space/")
+faceswap_client = Client("https://fffiloni-deepfakeai.hf.space/", hf_token=hf_token)
 
 def infer(portrait_in, prompt_in):
     # Generate Image from SDXL
@@ -12,12 +15,19 @@ def infer(portrait_in, prompt_in):
         api_name="/infer"
     )
 
+    unique_id = str(uuid.uuid4())
+    
     # Face Swap
-    gr.Info("Face swap your face on result ... This can take a while depending on @tonyassi Face Swap demo queue.")
+    gr.Info("Face swap your face on result ...")
     faceswap_result = faceswap_client.predict(
-    	portrait_in,
-        sdxl_result,
-        api_name="/predict"
+        portrait_in,	# str (filepath or URL to image) in 'SOURCE IMAGE' Image component
+        sdxl_result,	# str (filepath or URL to image) in 'TARGET IMAGE' Image component
+        unique_id,	# str in 'parameter_12' Textbox component
+        ["face_swapper"],	# List[str] in 'FRAME PROCESSORS' Checkboxgroup component
+        "left-right",	# str (Option from: ['left-right', 'right-left', 'top-bottom', 'bottom-top', 'small-large', 'large-small']) in 'FACE ANALYSER DIRECTION' Dropdown component
+        "none",	# str (Option from: ['none', 'reference', 'many']) in 'FACE RECOGNITION' Dropdown component
+        "none",	# str (Option from: ['none', 'male', 'female']) in 'FACE ANALYSER GENDER' Dropdown component
+        fn_index=1
     )
 
     return faceswap_result