Update README.md

README.md

@@ -11,18 +11,22 @@ tags:
 
 这个模型是一个旨在生成 electronic 风格音乐的非条件性扩散模型
 
-# Usage : 
-
-
-from diffusers import DDPMPipeline
+## Usage
 
+```python
+import torch 
+from diffusers import DiffusionPipeline
 from IPython.display import Audio
 
 
-pipe = DDPMPipeline.from_pretrained('Kevin3111/Electronic_test')
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"device: {device}")
 
-output = pipe(num_inference_steps=50)
+pipe = DiffusionPipeline.from_pretrained(
+    'Kevin3111/Electronic_test'
+).to(device)
+output = pipe(steps=50)
 
 display(output.images[0])
-
 display(Audio(output.audios[0], rate=pipe.mel.get_sample_rate()))
+```