Update app.py

app.py

@@ -9,7 +9,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 def infer(prompt):
 
     prompt = "a photo of an astronaut riding a horse on mars"
-    image = pipe(prompt).images[0]  
+    image = pipe(prompt=prompt, guidance_scale=10.0, num_inference_steps=12, width=256, height=256).images[0]  
 
     return image