stylestudio

app.py

@@ -1,6 +1,7 @@
 import sys
 sys.path.append("./")
 import gradio as gr
+import spaces
 import torch
 from ip_adapter.utils import BLOCKS as BLOCKS
 import numpy as np
@@ -84,7 +85,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
-
+@spaces.GPU
 def create_image(
                  style_image_pil,
                  prompt,