Commit
·
5639c9e
1
Parent(s):
8ca826d
Fix path issue
Browse files- synthesize.py +1 -1
synthesize.py
CHANGED
@@ -18,7 +18,7 @@ from text import text_to_sequence
|
|
18 |
|
19 |
from transformers import RobertaTokenizerFast, AutoModel, AutoModelForSequenceClassification
|
20 |
|
21 |
-
ro_model = "
|
22 |
roberta_tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')
|
23 |
roberta_model = AutoModelForSequenceClassification.from_pretrained(ro_model)
|
24 |
|
|
|
18 |
|
19 |
from transformers import RobertaTokenizerFast, AutoModel, AutoModelForSequenceClassification
|
20 |
|
21 |
+
ro_model = "roberta_pretrained"
|
22 |
roberta_tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')
|
23 |
roberta_model = AutoModelForSequenceClassification.from_pretrained(ro_model)
|
24 |
|