Hugging Face's logo

solwol
/
roberta-sentiment-classifier-peft

Text Classification
PEFT
Safetensors
roberta
Model card Files Community
solwol commited on
Commit
2b95880
·
verified ·
1 Parent(s): 2b9be0b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ from transformers import AutoModelForSequenceClassification, RobertaConfig
28
 
29
  config = RobertaConfig.from_pretrained(
30
  "roberta-base",
31
- id2label={ 0: "👎", 1: "👍"}
32
  )
33
  model = AutoModelForSequenceClassification.from_pretrained("roberta-base", config=config)
34
  model = PeftModel.from_pretrained(model, "solwol/roberta-sentiment-classifier-peft")
 
28
 
29
  config = RobertaConfig.from_pretrained(
30
  "roberta-base",
31
+ id2label={0: "👎", 1: "👍"}
32
  )
33
  model = AutoModelForSequenceClassification.from_pretrained("roberta-base", config=config)
34
  model = PeftModel.from_pretrained(model, "solwol/roberta-sentiment-classifier-peft")