Update app.py

app.py

@@ -18,11 +18,14 @@ def generate_image(prompt):
         if not hasattr(scheduler_step_storage, 'step'):
             scheduler_step_storage.step = 0
         
+        # Use a sensible default for num_inference_steps
+        num_inference_steps = 1  # Adjust this value as needed
+        
         # Use the thread-local step index
         output = model(
             prompt=prompt, 
-            num_inference_steps=1,  # Set to a more reasonable default for inference steps
-            guidance_scale=0.0,  # Adjust guidance scale for better image quality
+            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
         )