Update app.py

app.py

@@ -1,7 +1,7 @@
 
 import pandas as pd
 import torch
-from sklearn.model_selection import train_test_split
+
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
 
 # Load the pre-trained BERT model and tokenizer