Hugging Face's logo

Spaces:
frugal-ai-challenge
/
text-jennasparks
Sleeping

App Files Community
jennasparks commited on
Commit
fc31f61
·
verified ·
1 Parent(s): 660641a

tokenizer separate

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -30,7 +30,7 @@ ROUTE = "/text"
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)
34
 
35
 
36
  @router.post(ROUTE, tags=["Text Task"],
 
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("google/electra-base-discriminator")
34
 
35
 
36
  @router.post(ROUTE, tags=["Text Task"],