Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,7 @@ from transformers import RobertaForSequenceClassification, RobertaTokenizer
|
|
| 29 |
import torch
|
| 30 |
|
| 31 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
| 32 |
-
model = RobertaForSequenceClassification.from_pretrained("
|
| 33 |
|
| 34 |
sentence = "Your input sentence here."
|
| 35 |
inputs = tokenizer(sentence, return_tensors="pt")
|
|
|
|
| 29 |
import torch
|
| 30 |
|
| 31 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
| 32 |
+
model = RobertaForSequenceClassification.from_pretrained("oeg/RoBERTa-Repository-Proposal")
|
| 33 |
|
| 34 |
sentence = "Your input sentence here."
|
| 35 |
inputs = tokenizer(sentence, return_tensors="pt")
|