update

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import spaces 
+import subprocess
 import torch
 import torchvision
 from torchvision import transforms
@@ -66,6 +67,14 @@ from utils.infer_utils import remove_background, resize_foreground
 
 SEED = 0
 
+def install_dependency():
+    # install apex
+    subprocess.run(
+        f'{sys.executable} -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git@https://github.com/NVIDIA/apex.git',
+        shell=True,
+    )
+
+
 def resize_to_224(img):
     img = transforms.functional.resize(img, 224,
         interpolation=transforms.InterpolationMode.LANCZOS)
@@ -451,6 +460,8 @@ if __name__ == "__main__":
 
     args.rendering_kwargs = rendering_options_defaults(args)
 
+    install_dependency()
+
     # main(args)
 
     zero = torch.Tensor([0]).cuda()