Update app.py

app.py

@@ -4,10 +4,11 @@ os.system('pip install transformers')
 import os
 os.system('pip install torch')
 # Import the necessary libraries
+# Import the necessary libraries
 from transformers import AutoModel, AutoTokenizer
 import torch
 from torch.utils.data import DataLoader, Dataset
-from sklearn.model_selection import train_test_split  # Importing train_test_split from scikit-learn
+from sklearn.model_selection import train_test_split  # Corrected import statement
 import pandas as pd
 import gradio as gr