KingNish commited on
Commit
a25ba7a
1 Parent(s): f1c2277

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)
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)
@@ -56,7 +56,7 @@ examples = [
56
  with gr.Blocks() as demo:
57
  with gr.Column(elem_id="app-container"):
58
  gr.Markdown("# 🎨 Realtime FLUX Image Generator")
59
- gr.Markdown("Generate stunning images in real-time with advanced AI technology.")
60
  gr.Markdown("<span style='color: red;'>Note: Sometimes it stucks or stops generating images (I don't know why). In that situation just refresh the site.</span>")
61
 
62
  with gr.Row():
 
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)
 
56
  with gr.Blocks() as demo:
57
  with gr.Column(elem_id="app-container"):
58
  gr.Markdown("# 🎨 Realtime FLUX Image Generator")
59
+ gr.Markdown("Generate stunning images in real-time with Modified Flux.Schnell pipeline.")
60
  gr.Markdown("<span style='color: red;'>Note: Sometimes it stucks or stops generating images (I don't know why). In that situation just refresh the site.</span>")
61
 
62
  with gr.Row():