Update README.md

README.md

@@ -36,18 +36,17 @@ license: cc-by-nc-4.0
 
 ## Usage
 ``` python
-from transformers import RobertaTokenizerFast, TFRobertaForSequenceClassification, TextClassificationPipeline
+from transformers import RobertaTokenizerFast, RobertaForSequenceClassification, TextClassificationPipeline
 
 # Load fine-tuned MRC model by HuggingFace Model Hub
 HUGGINGFACE_MODEL_PATH = "bespin-global/klue-roberta-small-3i4k-intent-classification"
-loaded_tokenizer = RobertaTokenizerFast.from_pretrained(MODEL_SAVE_PATH)
-loaded_model = TFRobertaForSequenceClassification.from_pretrained(MODEL_SAVE_PATH)
+loaded_tokenizer = RobertaTokenizerFast.from_pretrained(HUGGINGFACE_MODEL_PATH )
+loaded_model = RobertaForSequenceClassification.from_pretrained(HUGGINGFACE_MODEL_PATH )
 
 # using Pipeline
 text_classifier = TextClassificationPipeline(
     tokenizer=loaded_tokenizer, 
     model=loaded_model, 
-    framework='tf',
     return_all_scores=True
 )