Update README.md

README.md

@@ -64,4 +64,23 @@ def predict(text):
     return probs, pred_label_idx, pred_label
 ```
 
+- How to infer the model in huggingface ?
+
+```
+model_path = "KN123/PromptClassifier-v1.0"
+
+
+model = BertForSequenceClassification.from_pretrained(model_path)
+tokenizer= BertTokenizerFast.from_pretrained(model_path)
+nlp= pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+```
+
+```
+print(predict("set the color Red to items F-3333"))
+(tensor([[0.9979, 0.0021]], grad_fn=<SoftmaxBackward0>), tensor(0), 'instruction_prompt')
+
+nlp("Calculate the average attendance at each sports stadium.")
+(tensor([[0.0038, 0.9962]], grad_fn=<SoftmaxBackward0>), tensor(1), 'sql_prompt')
+```
+