Update README.md

README.md

@@ -26,11 +26,10 @@ pip install git+https://github.com/jimbozhang/hf_transformers_custom_model_dashe
 ```python
 >>> model_name = "mispeech/dasheng-base"
 
->>> from dasheng_model.feature_extraction_dasheng import DashengFeatureExtractor
->>> from dasheng_model.modeling_dasheng import DashengModel
+>>> from transformers import AutoModel, AutoFeatureExtractor
 
->>> feature_extractor = DashengFeatureExtractor.from_pretrained(model_name)
->>> model = DashengModel.from_pretrained(model_name, outputdim=None)  # no linear output layer if `outputdim` is `None`
+>>> feature_extractor = AutoFeatureExtractor.from_pretrained(model_name, trust_remote_code=True)
+>>> model = AutoModel.from_pretrained(model_name, outputdim=None, trust_remote_code=True)
 
 >>> inputs = feature_extractor(torch.randn(1, 16000), sampling_rate=sampling_rate, return_tensors="pt")
 >>> inputs.input_values.shape