Update README.md
Browse files
README.md
CHANGED
|
@@ -68,7 +68,12 @@ I have fine-tuned ParsBert for classifying Persian text using the "Msobhi/virgoo
|
|
| 68 |
Utilize this code snippet to test the model. It may not be very accurate, but further training epochs could enhance its performance.
|
| 69 |
|
| 70 |
model_path = "Aminrhmni/Persian-text-classification-model"
|
|
|
|
| 71 |
from transformers import pipeline, BertForSequenceClassification, BertTokenizerFast
|
|
|
|
| 72 |
model = BertForSequenceClassification.from_pretrained(model_path)
|
|
|
|
| 73 |
tokenizer= BertTokenizerFast.from_pretrained(model_path)
|
|
|
|
| 74 |
nlp= pipeline("text-classification", model=model, tokenizer=tokenizer)
|
|
|
|
|
|
| 68 |
Utilize this code snippet to test the model. It may not be very accurate, but further training epochs could enhance its performance.
|
| 69 |
|
| 70 |
model_path = "Aminrhmni/Persian-text-classification-model"
|
| 71 |
+
|
| 72 |
from transformers import pipeline, BertForSequenceClassification, BertTokenizerFast
|
| 73 |
+
|
| 74 |
model = BertForSequenceClassification.from_pretrained(model_path)
|
| 75 |
+
|
| 76 |
tokenizer= BertTokenizerFast.from_pretrained(model_path)
|
| 77 |
+
|
| 78 |
nlp= pipeline("text-classification", model=model, tokenizer=tokenizer)
|
| 79 |
+
|