burningfalls
/

my-fine-tuned-bert

Text Classification

Inference Endpoints

Model card Files Files and versions Community

burningfalls commited on May 27, 2023

Commit

c2ba529

•

1 Parent(s): 0a5b46b

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -40,6 +40,8 @@ transformers>=4.0
 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
 from transformers import TextClassificationPipeline
 def load_bert():
     loaded_tokenizer = AutoTokenizer.from_pretrained(BERT_PATH)
     loaded_model = TFAutoModelForSequenceClassification.from_pretrained(BERT_PATH)

 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
 from transformers import TextClassificationPipeline
+BERT_PARH = "burningfalls/my-fine-tuned-bert"
 def load_bert():
     loaded_tokenizer = AutoTokenizer.from_pretrained(BERT_PATH)
     loaded_model = TFAutoModelForSequenceClassification.from_pretrained(BERT_PATH)