jennasparks commited on
Commit
660641a
·
verified ·
1 Parent(s): 6e88853

Error in repo name

Browse files
Files changed (1) hide show
  1. 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/electra-tf"
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)