update-model-card

README.md

@@ -4,6 +4,7 @@ license: apache-2.0
 tags:
 - text-classfication
 - int8
+- neural-compressor
 - Intel® Neural Compressor
 - PostTrainingStatic
 datasets:
@@ -46,18 +47,20 @@ The same model is provided in two different formats: PyTorch and ONNX.
 | Primary intended users | Anyone | 
 | Out-of-scope uses | This model is already fine-tuned and quantized to INT8. It is not suitable for further fine-tuning in this form. To fine-tune your own model, you can start with [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english). |
 
-#### Load PyTorch model with Optimum
+#### Load the PyTorch model with Optimum Intel
 ```python
-from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSequenceClassification
-int8_model = IncQuantizedModelForSequenceClassification.from_pretrained(
-    'Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static')
+from optimum.intel.neural_compressor import INCModelForSequenceClassification
+
+model_id = "Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static"
+int8_model = INCModelForSequenceClassification.from_pretrained(model_id)
 ```
 
-#### Load ONNX model:
+#### Load the ONNX model with Optimum:
 ```python
 from optimum.onnxruntime import ORTModelForSequenceClassification
-model = ORTModelForSequenceClassification.from_pretrained(
-    'Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static')
+
+model_id = "Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static"
+int8_model = ORTModelForSequenceClassification.from_pretrained(model_id)
 ```
 
 | Factors | Description | 
@@ -109,4 +112,4 @@ model = ORTModelForSequenceClassification.from_pretrained(
   year      = {2022},
   url       = {https://huggingface.co/Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static},
 }
-```
+```