mrcuddle commited on
Commit
bd7219f
1 Parent(s): a82bd01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -92,8 +92,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
92
 
93
  with gr.Column(scale=4):
94
  output = gr.Image(label="Your Generated Image")
95
-
96
- @spaces.GPU
97
  def process_image(height, width, steps, scales, prompt, seed, preset_prompt):
98
  global pipe
99
  with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
 
92
 
93
  with gr.Column(scale=4):
94
  output = gr.Image(label="Your Generated Image")
95
+
96
+ @spaces.GPU
97
  def process_image(height, width, steps, scales, prompt, seed, preset_prompt):
98
  global pipe
99
  with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):