Update app.py

app.py

@@ -8,9 +8,9 @@ import tensorflow as tf
 import transformers
 from transformers import pipeline
 from transformers import RobertaTokenizer, RobertaModel
-from transformers import AutoModelForSequenceClassification
+from transformers import AutoModelForSequenceClassification, AutoModelForTokenClassification
 from transformers import TFAutoModelForSequenceClassification
-from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers import AutoTokenizer
 
 import matplotlib.pyplot as plt
 import sys
@@ -20,7 +20,7 @@ csv.field_size_limit(sys.maxsize)
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-tokenizer = AutoTokenizer.from_pretrained("jschwaller/ADRv2024")
+tokenizer = AutoTokenizer.from_pretrained("jschwaller/ADRv2024")  
 model = AutoModelForSequenceClassification.from_pretrained("jschwaller/ADRv2024")
 
 # build a pipeline object to do predictions