Spaces:

alex-abb
/

LLM_Feeling_Analyzer

Runtime error

alex-abb commited on Jun 21

Commit

9277ff4

•

1 Parent(s): 5a974a6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import datasets
 from datasets import load_dataset
 import transformers
-from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
-model = TFAutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
 tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 model = BertForSequenceClassification.from_pretrained(model_name, num_labels=2)  # 2 classes : positif et négatif

 import datasets
 from datasets import load_dataset
 import transformers
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, Trainer, TrainingArguments, pipeline
+model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)  # 2 classes : positif et négatif
 tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 model = BertForSequenceClassification.from_pretrained(model_name, num_labels=2)  # 2 classes : positif et négatif