hf spaces

app.py

@@ -11,8 +11,6 @@ import os
 import numpy as np
 from PIL import Image
 
-import spaces
-
 path = os.getcwd()
 output_dir = f"{path}/output"
 input_dir = f"{path}/input"
@@ -25,7 +23,6 @@ class webui:
     def __init__(self):
         self.demo = gr.Blocks()
 
-    @spaces.GPU
     def undercoat(self, input_image, pos_prompt, neg_prompt, alpha_th):
         org_line_image = input_image
         image = pil2cv(input_image)

sd_model.py

@@ -2,6 +2,8 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCM
 from diffusers import StableDiffusionXLControlNetPipeline, ControlNetModel, AutoencoderKL
 import torch
 
+import spaces
+
 device = "cuda"
 
 def get_cn_pipeline():
@@ -48,6 +50,7 @@ def get_cn_detector(image):
     print(detectors)
     return detectors
 
+@spaces.GPU
 def generate(pipe, detectors, prompt, negative_prompt):
     default_pos = "1girl, bestquality, 4K, ((white background)), no background"
     default_neg = "shadow, (worst quality, low quality:1.2), (lowres:1.2), (bad anatomy:1.2), (greyscale, monochrome:1.4)"