Update app.py

app.py

@@ -2,14 +2,19 @@ from diffusers import StableDiffusionXLPipeline
 import torch
 import gradio as gr
 
-pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
+pipe = StableDiffusionXLPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    torch_dtype=torch.float16,
+    use_safetensors=True,
+    variant="fp16"
+)
 pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
 def greet(prompt):
     print(prompt)
-    images = pipe(prompt=prompt).images[0]
-    print(images)
-    return images
+    image = pipe(prompt).images[0]
+    print(image)
+    return image
 
 demo = gr.Interface(
     fn=greet,