Update app.py

app.py

@@ -5,12 +5,13 @@ from diffusers import StableDiffusionPipeline
 from generate_propmts import generate_prompt
 
 # Load the model pipeline
-pipeline = StableDiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo").to("cuda")
+model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 
 async def generate_image(prompt, prompt_name):
     try:
         print(f"Generating image for {prompt_name}")
-        image = await pipeline(prompt).images[0]
+        output = await (model, prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
+        image = output.images[0]
         img_bytes = image.tobytes()
         print(f"Image bytes length for {prompt_name}: {len(img_bytes)}")
         return img_bytes