Update README.md

README.md

@@ -18,14 +18,21 @@ pip install diffusers transformers accelerate scipy safetensors
 Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
 
 ```python
+import torch
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+
 model_id = "lcybuaa/Text2Earth"
-# Maybe you can use the Euler scheduler here instead
+# Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to DPMSolverMultistepScheduler):
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16, safety_checker=None)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, scheduler=scheduler, 
+                                         custom_pipeline="pipeline_text2earth_diffusion", safety_checker=None)
 pipe = pipe.to("cuda")
 prompt = "Seven green circular farmlands are neatly arranged on the ground"
-image = pipe(prompt).images[0]
-    
+image = pipe(prompt,
+             height=256,
+             width=256,
+             num_inference_steps=50, 
+             guidance_scale=4.0).images[0]
+
 image.save("circular.png")
 ```