Update README.md

README.md

@@ -45,21 +45,23 @@ The `diffusion_pytorch_model_onlinehps.safetensors` is online version of SRPO ba
 #### Inference
 Replace the `ddiffusion_pytorch_model_onlinehps.safetensors` of FLUX
 ```python
-  pipe = FluxPipeline.from_pretrained('your dir',
-          torch_dtype=torch.bfloat16,
-          use_safetensors=True
-      ).to("cuda")
-  state_dict = load_file("yourpath")
-  pipe.transformer.load_state_dict(state_dict)
-  image = pipe(
-      prompt,
-      guidance_scale=3.5,
-      height=1024,
-      width=1024,
-      num_inference_steps=infer_step,
-      max_sequence_length=512,
-      generator=generator
-  ).images[0]
+from diffusers import FluxPipeline
+prompt='The Death of Ophelia by John Everett Millais, Pre-Raphaelite painting, Ophelia floating in a river surrounded by flowers, detailed natural elements, melancholic and tragic atmosphere'
+pipe = FluxPipeline.from_pretrained('./data/flux',
+        torch_dtype=torch.bfloat16,
+        use_safetensors=True
+    ).to("cuda")
+state_dict = load_file("./srpo/diffusion_pytorch_model.safetensors")
+pipe.transformer.load_state_dict(state_dict)
+image = pipe(
+    prompt,
+    guidance_scale=3.5,
+    height=1024,
+    width=1024,
+    num_inference_steps=infer_step,
+    max_sequence_length=512,
+    generator=generator
+).images[0]
 ```
 ### License
 SRPO is licensed under the License Terms of SRPO. See `./License.txt` for more details.