Update app.py

app.py

@@ -1,20 +1,23 @@
 import gradio as gr
-from diffusers import AutoPipelineForText2Image
+from diffusers import AutoPipelineForText2Image, EulerAncestralDiscreteScheduler
 from generate_propmts import generate_prompt
 from PIL import Image
 import asyncio
 import traceback
 
-# Load the model once outside of the function
+# Load the model with a modified 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
+
 
 async def generate_image(prompt):
     try:
-        num_inference_steps = 4  # Increased inference steps
+        num_inference_steps = 25  # Or your desired value
         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"
         )
@@ -26,7 +29,7 @@ async def generate_image(prompt):
         print(f"Error generating image: {e}")
         traceback.print_exc()
         return None
-
+ 
 
 async def inference(sentence_mapping, character_dict, selected_style):
     images = []