Update app.py

app.py

@@ -4,6 +4,8 @@ from diffusers import DiffusionPipeline, AutoPipelineForText2Image
 import base64
 from io import BytesIO
 
+
+
 def load_amused_model():
     # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO") 
     # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo"
@@ -14,7 +16,8 @@ def generate_image(prompt):
     try:
         pipe = load_amused_model()
         generator = torch.Generator().manual_seed(8)  # Create a generator for reproducibility
-        image = pipe(prompt, generator=generator).images[0]  # Generate image from prompt
+        #image = pipe(prompt, generator=generator).images[0]  # Generate image from prompt
+        image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
         return image, None
     except Exception as e:
         return None, str(e)