Update app.py

app.py

@@ -9,7 +9,7 @@ pipe = None
 def load_model():
     global pipe
     pipe = StableDiffusionPipeline.from_pretrained(
-        "lambdalabs/miniSD-diffusers",
+        "Lykon/dreamshaper-8",
         torch_dtype=torch.float32,
         safety_checker=None
     )
@@ -26,7 +26,7 @@ def generate():
     try:
         data = request.json
         prompt = data.get("prompt", "a beautiful landscape")
-        image = pipe(prompt, num_inference_steps=6, height=256, width=256).images[0]
+        image = pipe(prompt, num_inference_steps=20, height=512, width=512).images[0]
         buf = io.BytesIO()
         image.save(buf, format="PNG")
         img_b64 = base64.b64encode(buf.getvalue()).decode()