hf spaces

app.py

@@ -11,6 +11,7 @@ import os
 import numpy as np
 from PIL import Image
 
+import spaces
 
 path = os.getcwd()
 output_dir = f"{path}/output"
@@ -24,6 +25,7 @@ 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)