A5hbr1ng3r
/

COLD-LoL

Model card Files Files and versions Community

A5hbr1ng3r commited on Mar 25, 2024

Commit

c7ba399

·

verified ·

1 Parent(s): 7627d40

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model_name = "thu-coai/roberta-base-cold"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)
-model_path = "finetuned_cold_LoL.pth"
 model.load_state_dict(torch.load(model_path))

 model_name = "thu-coai/roberta-base-cold"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)
+model_path = "finetuned_cold_LoL.pth" # Could be downloaded in this repo.
 model.load_state_dict(torch.load(model_path))