Walmart-the-bag commited on
Commit
b3e1ee8
β€’
1 Parent(s): c4cc804

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.flo
9
  pipe = pipe.to("cuda")
10
 
11
  @spaces.GPU()
12
- def text_to_image(prompt):
13
  image = pipe(prompt).images [0]
14
  return image
15
 
 
9
  pipe = pipe.to("cuda")
10
 
11
  @spaces.GPU()
12
+ def text_to_image(prompt, progress=gr.Progress(track_tqdm=True)):
13
  image = pipe(prompt).images [0]
14
  return image
15