update code

app.py

@@ -91,7 +91,6 @@ def install_cuda_toolkit():
 
 # 🔧 Set CUDA_HOME before anything else
 
-
 # Configure logging settings
 logging.basicConfig(
     level=logging.INFO,
@@ -593,7 +592,7 @@ def assert_input_image(input_image):
     if input_image is None:
         raise gr.Error("No image selected or uploaded!")
 
-
+@spaces.GPU(duration=100)
 def process_image(input_image, source_type, is_style, save_dir):
     """ 🎯 处理 input_image，根据是否是示例图片执行不同逻辑 """
     process_img_input_dir = os.path.join(save_dir, 'input_image')
@@ -678,7 +677,6 @@ css = """
 """
 
 def launch_gradio_app():
-    install_cuda_toolkit()
     styles = {
         "Ghibli": "Ghibli style avatar, anime style",
         "Pixar": "a 3D render of a face in Pixar style",
@@ -940,11 +938,26 @@ def launch_gradio_app():
 
         demo.queue()
         demo.launch(server_name="0.0.0.0")
+def _get_output(cmd):
+    try:
+        return subprocess.check_output(cmd).decode("utf-8")
+    except Exception as ex:
+        logging.exception(ex)
 
+    return None
 
 if __name__ == '__main__':
     import torch.multiprocessing as mp
     mp.set_start_method('spawn', force=True)
+    logging.info("Environment Variables: %s" % os.environ)
+    logging.info("Installing CUDA extensions...")
+    if _get_output(["nvcc", "--version"]) is None:
+        logging.info("Installing CUDA toolkit...")
+        install_cuda_toolkit()
+    else:
+        logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
+
+
     launch_pretrained()
     image_folder = "./demo_data/source_img/img_generate_different_domain/images512x512/demo_imgs"
     example_img_names = os.listdir(image_folder)