Hugging Face's logo

appohfaiths
/
dummy-model

Text Classification
Transformers
Safetensors
distilbert
text-embeddings-inference
Model card Files
xet
Community
appohfaiths commited on
Commit
eb631d9
·
1 Parent(s): bb9762b

updates

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassification,DistilBertConfig
2
 
3
- config = DistilBertConfig.from_pretrained("./config.json", id2label=id2label, label2id=label2id)
4
  model = DistilBertForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english",config=config)
5
  tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english")
6
 
 
1
  from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassification,DistilBertConfig
2
 
3
+ config = DistilBertConfig.from_pretrained("./config.json")
4
  model = DistilBertForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english",config=config)
5
  tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english")
6