Update code snippet

README.md

@@ -27,14 +27,16 @@ tags:
 
 ## How to Get Started With the Model
 
+This requires to install Optimum by doing `pip install optimum[neural-compressor]`
+
 To load the quantized model and run inference using the Transformers [pipelines](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines), you can do as follows:
 
 ```python
 from transformers import AutoTokenizer, pipeline
-from optimum.intel.neural_compressor import IncQuantizedModelForSequenceClassification
+from optimum.intel import INCModelForSequenceClassification
 
 model_id = "echarlaix/distilbert-sst2-inc-dynamic-quantization-magnitude-pruning-0.1"
-model = IncQuantizedModelForSequenceClassification.from_pretrained(model_id)
+model = INCModelForSequenceClassification.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 cls_pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 text = "He's a dreadful magician."