Update app.py

app.py

@@ -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):
+def text_to_image(prompt, progress=gr.Progress(track_tqdm=True)):
  image = pipe(prompt).images [0]
  return image