Update app.py

app.py

@@ -5,19 +5,18 @@ from PIL import Image
 import asyncio
 import traceback
 
-# Load the model with a modified scheduler
+# Load the model with a different scheduler
 model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
-model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
-model.scheduler.config.prediction_type = "epsilon"  # Adjust prediction_type
+model.scheduler = DPMSolverMultistepScheduler.from_config(model.scheduler.config)  # Changed scheduler
 
 
 async def generate_image(prompt):
     try:
-        num_inference_steps = 5  # Or your desired value
+        num_inference_steps = 5  # You can adjust this
         output = await asyncio.to_thread(
             model,
             prompt=prompt,
-            num_inference_steps=num_inference_steps,  
+            num_inference_steps=num_inference_steps, 
             guidance_scale=0.0,
             output_type="pil"
         )