update code snippet

README.md

@@ -10,6 +10,7 @@ tags:
 - text-classification
 - neural-compressor
 - int8
+- 8-bit
 ---
 
 # Dynamically quantized DistilBERT base uncased finetuned SST-2
@@ -27,14 +28,17 @@ tags:
 
 ## How to Get Started With the Model
 
+This requires to install Optimum :
+`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-base-uncased-finetuned-sst-2-english-int8-dynamic"
-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."