Update app.py

app.py

@@ -2,12 +2,12 @@ import os
 import asyncio
 import concurrent.futures
 from io import BytesIO
-from diffusers import AutoPipelineForText2Image
+from diffusers import StableDiffusionPipeline
 import gradio as gr
 from generate_prompts import generate_prompt
 
 # Initialize model globally
-model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
+model = StableDiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
 
 def generate_image(prompt, prompt_name):
     """
@@ -20,7 +20,7 @@ def generate_image(prompt, prompt_name):
     """
     try:
         print(f"Generating image for {prompt_name}")
-        output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
+        output = model(prompt=prompt, num_inference_steps=50, guidance_scale=7.5)
         if isinstance(output.images, list) and len(output.images) > 0:
             image = output.images[0]
             buffered = BytesIO()