Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import os
 import numpy as np
@@ -108,6 +109,7 @@ def load_b_loras(content_b_lora, style_b_lora):
     
     return prepared_prompt
 
+@spaces.GPU()
 def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed: