Update app.py

app.py

@@ -10,28 +10,23 @@ model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 
 async def generate_image(prompt):
     try:
-        # Use a higher value for num_inference_steps to avoid IndexError
-        num_inference_steps = 3  # Adjust this value as needed
-
-        # Use the model to generate an image
+        num_inference_steps = 4  # Increased inference steps
         output = await asyncio.to_thread(
             model,
             prompt=prompt,
-            num_inference_steps=num_inference_steps,  # Use a higher value for inference steps
-            guidance_scale=0.0,  # Typical value for guidance scale in image generation
-            output_type="pil"  # Directly get PIL Image objects
+            num_inference_steps=num_inference_steps,
+            guidance_scale=0.0,
+            output_type="pil"
         )
-
-        # Check for output validity and return
         if output.images:
             return output.images[0]
         else:
             raise Exception("No images returned by the model.")
-    
     except Exception as e:
         print(f"Error generating image: {e}")
         traceback.print_exc()
-        return None  # Return None on error to handle it gracefully in the UI
+        return None
+
 
 async def inference(sentence_mapping, character_dict, selected_style):
     images = []