Spaces:

Walmart-the-bag
/

Juggernaut-X-v10

Running on Zero

Walmart-the-bag commited on Apr 25, 2024

Commit

b3e1ee8

verified ·

1 Parent(s): c4cc804

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.flo
 pipe = pipe.to("cuda")
 @spaces.GPU()
-def text_to_image(prompt):
  image = pipe(prompt).images [0]
  return image

 pipe = pipe.to("cuda")
 @spaces.GPU()
+def text_to_image(prompt, progress=gr.Progress(track_tqdm=True)):
  image = pipe(prompt).images [0]
  return image