akameswa commited on
Commit
8110449
1 Parent(s): 29a0427

Update src/pipelines/negative.py

Browse files
Files changed (1) hide show
  1. src/pipelines/negative.py +2 -1
src/pipelines/negative.py CHANGED
@@ -1,8 +1,9 @@
 
1
  import gradio as gr
2
  from src.util.base import *
3
  from src.util.params import *
4
 
5
-
6
  def display_negative_images(
7
  prompt, seed, num_inference_steps, negative_prompt="", progress=gr.Progress()
8
  ):
 
1
+ import spaces
2
  import gradio as gr
3
  from src.util.base import *
4
  from src.util.params import *
5
 
6
+ @spaces.GPU(enable_queue=True)
7
  def display_negative_images(
8
  prompt, seed, num_inference_steps, negative_prompt="", progress=gr.Progress()
9
  ):