fix

app.py

@@ -13,6 +13,7 @@ from random import randint
 from utils import init_latent
 from transformers import Blip2Processor, Blip2ForConditionalGeneration
 from diffusers import DDIMScheduler
+import spaces
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 if device == 'cpu':
@@ -147,6 +148,7 @@ def load_example_style():
 
     return examples
 
+@spaces.GPU
 def style_fn(image_path, style_name, content_text, output_number, diffusion_step=50):
     user_image_flag = not style_name.strip() # empty
 
@@ -219,7 +221,7 @@ def style_fn(image_path, style_name, content_text, output_number, diffusion_step
         use_advanced_sampling = False
         use_prompt_as_null = True
 
-        ref_prompt = blip_inf_prompt(origin_real_img).to(device)
+        ref_prompt = blip_inf_prompt(origin_real_img)
         inf_prompt = inf_object
         style_description_neg = None
 
@@ -275,6 +277,7 @@ def style_fn(image_path, style_name, content_text, output_number, diffusion_step
 
         return grid
 
+@spaces.GPU
 def controlnet_fn(image_path, depth_image_path, style_name, content_text, output_number, controlnet_scale=0.5, diffusion_step=50):
     config_path = './config/{}.json'.format(style_name)
     config = parse_config(config_path)