Update README.md

README.md

@@ -85,13 +85,15 @@ Speculative Decoding was proposed in Fast Inference from Transformers via Specul
 ```bash
 import torch
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor
+from transformers import pipeline
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-
+# load dataset
 dataset = load_dataset("bond005/sberdevices_golos_10h_crowd", split="validation", token=True)
 
+# load model
 model_id = "Val123val/ru_whisper_small"
 
 model = AutoModelForSpeechSeq2Seq.from_pretrained(
@@ -105,6 +107,7 @@ model.to(device)
 
 processor = AutoProcessor.from_pretrained(model_id)
 
+# load assistant model
 assistant_model_id = "openai/whisper-tiny"
 
 assistant_model = AutoModelForSpeechSeq2Seq.from_pretrained(
@@ -117,8 +120,8 @@ assistant_model = AutoModelForSpeechSeq2Seq.from_pretrained(
 
 assistant_model.to(device);
 
-from transformers import pipeline
 
+# make pipe
 pipe = pipeline(
     "automatic-speech-recognition",
     model=model,