Update app.py

app.py

@@ -26,9 +26,11 @@ def generate(prompt, option, progress=gr.Progress()):
     print(prompt, option)
     ckpt, step = opts[option]
 
+    progress(0, desc="Downloading the model")
+
     # Main pipeline.
     unet = UNet2DConditionModel.from_config(base, subfolder="unet")
-    pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=dtype, variant="fp16")
+    pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=dtype, variant="fp16").to(device, dtype)
     pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device))
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", prediction_type="sample" if step == 1 else "epsilon")