Update README.md

README.md

@@ -31,11 +31,11 @@ pip install git+https://github.com/jimbozhang/hf_transformers_custom_model_dashe
 >>> feature_extractor = AutoFeatureExtractor.from_pretrained(model_name, trust_remote_code=True)
 >>> model = AutoModel.from_pretrained(model_name, outputdim=None, trust_remote_code=True)
 
+>>> import torch
 >>> inputs = feature_extractor(torch.randn(1, 16000), sampling_rate=sampling_rate, return_tensors="pt")
 >>> inputs.input_values.shape
 torch.Size([1, 64, 101])   # 64 mel-filterbanks, 101 frames
 
->>> import torch
 >>> with torch.no_grad():
 ...     outputs = model(**inputs)