Error in repo name
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -27,7 +27,7 @@ ROUTE = "/text"
|
|
27 |
# model.load_weights(model_weights_path)
|
28 |
# tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
|
29 |
|
30 |
-
model_repo = "jennasparks/
|
31 |
config = ElectraConfig.from_pretrained(model_repo)
|
32 |
model = TFElectraForSequenceClassification.from_pretrained(model_repo)
|
33 |
tokenizer = ElectraTokenizer.from_pretrained(model_repo)
|
|
|
27 |
# model.load_weights(model_weights_path)
|
28 |
# tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
|
29 |
|
30 |
+
model_repo = "jennasparks/electra_tf"
|
31 |
config = ElectraConfig.from_pretrained(model_repo)
|
32 |
model = TFElectraForSequenceClassification.from_pretrained(model_repo)
|
33 |
tokenizer = ElectraTokenizer.from_pretrained(model_repo)
|