text-jennasparks

Sleeping

jennasparks commited on Feb 1

Commit

6e88853

verified ·

1 Parent(s): d0e440d

electra specific

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -28,9 +28,9 @@ ROUTE = "/text"
 # tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
 model_repo = "jennasparks/electra-tf"
-config = AutoConfig.from_pretrained(model_repo)
-model = TFAutoModelForSequenceClassification.from_pretrained(model_repo)
-tokenizer = AutoTokenizer.from_pretrained(model_repo)
 @router.post(ROUTE, tags=["Text Task"],

 # tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
 model_repo = "jennasparks/electra-tf"
+config = ElectraConfig.from_pretrained(model_repo)
+model = TFElectraForSequenceClassification.from_pretrained(model_repo)
+tokenizer = ElectraTokenizer.from_pretrained(model_repo)
 @router.post(ROUTE, tags=["Text Task"],