Update README.md

README.md

@@ -1,7 +1,8 @@
 # To generate synthetic ECGs
-
+```python
 from transformers import AutoModels
 
 model = AutoModels.from_pretrained("deepsynthbody/deepfake_ecg")
 
-out = model(num_channels=5)
+out = model(num_samples=5)
+```