KingNish commited on
Commit
7d90483
1 Parent(s): a25ba7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
22
  torch.cuda.empty_cache()
23
 
24
  # Inference function
25
- @spaces.GPU(duration=25, queue=False)
26
  def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
27
  if randomize_seed:
28
  seed = random.randint(0, MAX_SEED)
 
22
  torch.cuda.empty_cache()
23
 
24
  # Inference function
25
+ @spaces.GPU(duration=25)
26
  def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
27
  if randomize_seed:
28
  seed = random.randint(0, MAX_SEED)