Update app.py

app.py

@@ -1,8 +1,8 @@
-import pandas as pd
-from sklearn.model_selection import train_test_split
-from sklearn.preprocessing import StandardScaler
-from transformers import BertTokenizer, BertForSequenceClassification, Trainer, TrainingArguments
-import torch
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+
+tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
+model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-uncased")
 
 # Step 1: Data Collection and Preprocessing
 # Load the dataset