update spaces

app.py

@@ -2,13 +2,16 @@ import cv2
 import numpy as np
 from PIL import Image
 import os
+import spaces
 import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
 import shlex
 import subprocess
 
+subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 subprocess.run(shlex.split('pip install flash-attn --no-build-isolation'),
                env=os.environ | {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"})
+subprocess.run(shlex.split('pip install scepter --no-deps'))
 
 def resolve_hf_path(path):
     if isinstance(path, str) and path.startswith("hf://"):
@@ -50,7 +53,7 @@ from inference.registry import INFERENCES
 config_path = os.path.join("config", "ace_plus_fft.yaml")
 cfg = Config(load=True, cfg_file=config_path)
 # Instantiate the ACEInference object.
-ace_infer = ACEInference(cfg)
+ace_infer = INFERENCES.build(cfg) # ACEInference(cfg)
 
 def create_face_mask(pil_image):
     """
@@ -94,6 +97,7 @@ def create_face_mask(pil_image):
         print(f"Error: {e}")
         raise ValueError('A very specific bad thing happened.')
 
+@spaces.GPU(duration=80)
 def face_swap_app(target_img, face_img):
     if target_img is None or face_img is None:
         raise ValueError("Both a target image and a face image must be provided.")
@@ -108,7 +112,7 @@ def face_swap_app(target_img, face_img):
         reference_image=target_img,
         edit_image=face_img,
         edit_mask=edit_mask,
-        prompt="retain face",
+        prompt="maintain the facial features as much as possible",
         output_height=1024,
         output_width=1024,
         sampler='flow_euler',