Update app.py

app.py

@@ -3,14 +3,14 @@ import asyncio
 from concurrent.futures import ProcessPoolExecutor
 from io import BytesIO
 from PIL import Image
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline, AutoPipelineForText2Image
 import gradio as gr
 from generate_prompts import generate_prompt
 
 # Load the model once at the start
 print("Loading the Stable Diffusion model...")
 try:
-    model = StableDiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
+    model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
     print("Model loaded successfully.")
 except Exception as e:
     print(f"Error loading model: {e}")