Update README.md

README.md

@@ -36,7 +36,7 @@ from peft import PeftModel
 from transformers import pipeline
 modelname="SeyedAli/Persian-Text-Sentiment-Bert-LORA"
 tokenizer=AutoTokenizer.from_pretrained("HooshvareLab/bert-base-parsbert-uncased",model_max_length=100, add_special_tokens = True)
-model=AutoModelForSequenceClassification.from_pretrained("HooshvareLab/bert-base-parsbert-uncased",num_labels=2,id2label=id2label,label2id=label2id)
+model=AutoModelForSequenceClassification.from_pretrained("HooshvareLab/bert-base-parsbert-uncased")
 model = PeftModel.from_pretrained(model, modelname)
 pipe = pipeline("text-classification", model=model,tokenizer=tokenizer)
 pipe('خیلی کتاب خوبی بود')