Update app.py

app.py

@@ -1,15 +1,17 @@
 import gradio as gr
-from diffusers import StableDiffusionPipeline
+#from diffusers import StableDiffusionPipeline
+from diffuser import AutoPipelineForText2Image
 import torch
 
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None)
+#model_id = "runwayml/stable-diffusion-v1-5"
+model_id = "stabilityai/sdxl-turbo"
+pipe = AutoPipelineForText2Image.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None)
 
 
 def infer(prompt):
 
     prompt = "a photo of an astronaut riding a horse on mars"
-    image = pipe(prompt=prompt, guidance_scale=10.0, num_inference_steps=6, width=256, height=256).images[0]  
+    image = pipe(prompt=prompt, guidance_scale=10.0, num_inference_steps=2, width=256, height=256).images[0]  
 
     return image