Spaces:

vasevooo
/

NLP_project

Sleeping

vasevooo commited on Jun 9, 2023

Commit

e84abd6

1 Parent(s): f615a30

Update pages/imdb.py

Files changed (1) hide show

pages/imdb.py CHANGED Viewed

@@ -144,7 +144,7 @@ def main():
     config = AutoConfig.from_pretrained('distilbert-base-uncased', num_labels=2)
     automodel = AutoModelForSequenceClassification.from_config(config)
     input_tokens = autotoken(
         review,

     config = AutoConfig.from_pretrained('distilbert-base-uncased', num_labels=2)
     automodel = AutoModelForSequenceClassification.from_config(config)
+    autotoken = transformers.AutoTokenizer.from_pretrained('distilbert-base-uncased-finetuned-sst-2-english')
     input_tokens = autotoken(
         review,