Update app.py

app.py

@@ -8,7 +8,6 @@ os.system('pip install torch')
 from transformers import AutoModel, AutoTokenizer
 import torch
 from torch.utils.data import DataLoader, Dataset
-from sklearn.model_selection import train_test_split  # Corrected import statement
 import pandas as pd
 import gradio as gr