Spaces:

wsdwJohn1231
/

text-classification

Sleeping

wsdwJohn1231 commited on Feb 3

Commit

5723a7c

verified ·

1 Parent(s): 681ca4d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import (
     AutoTokenizer,
 )
-model_dir = "my-bert-model"
 config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
 tokenizer = AutoTokenizer.from_pretrained(model_dir)

     AutoTokenizer,
 )
+model_dir = "wsdwJohn1231/my-bert-model"
 config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
 tokenizer = AutoTokenizer.from_pretrained(model_dir)